Estou tentando acessar o índice de uma linha em uma função aplicada em todo o DataFrame
Pandas. Eu tenho algo assim:
df = pandas.DataFrame([[1,2,3],[4,5,6]], columns=['a','b','c'])
>>> df
a b c
0 1 2 3
1 4 5 6
e vou definir uma função que acessa elementos com uma determinada linha
def rowFunc(row):
return row['a'] + row['b'] * row['c']
Posso aplicá-lo assim:
df['d'] = df.apply(rowFunc, axis=1)
>>> df
a b c d
0 1 2 3 7
1 4 5 6 34
Impressionante! Agora, e se eu quiser incorporar o índice à minha função? O índice de qualquer linha dada DataFrame
antes de adicionar d
seria Index([u'a', u'b', u'c', u'd'], dtype='object')
, mas eu quero 0 e 1. Portanto, não posso simplesmente acessar row.index
.
Eu sei que poderia criar uma coluna temporária na tabela onde armazeno o índice, mas estou me perguntando se ela está armazenada no objeto de linha em algum lugar.
apply
? É muito mais lento do que realizar operações vetorizadas no próprio quadro. (Às vezes, aplicar é a maneira mais simples de fazer algo, e as considerações de desempenho costumam ser exageradas, mas para o seu exemplo específico é tão fácil de não usar.)