lấy chỉ mục của một hàng trong gấu trúc áp dụng hàm

Question

Tôi đang cố gắng truy cập chỉ mục của một hàng trong một hàm được áp dụng trên toàn bộ DataFrame trong Pandas. Tôi có một cái gì đó như thế này:

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 

và tôi sẽ định nghĩa một hàm truy cập vào các yếu tố với một hàng cho

def rowFunc(row): 
    return row['a'] + row['b'] * row['c'] 

tôi có thể áp dụng nó như vậy:

df['d'] = df.apply(rowFunc, axis=1) 
>>> df 
    a b c d 
0 1 2 3 7 
1 4 5 6 34 

Awesome! Bây giờ nếu tôi muốn kết hợp chỉ mục vào chức năng của mình thì sao? Chỉ mục của bất kỳ hàng cụ thể nào trong số DataFrame trước khi thêm d sẽ là Index([u'a', u'b', u'c', u'd'], dtype='object'), nhưng tôi muốn 0 và 1. Vì vậy, tôi không thể truy cập row.index.

Tôi biết tôi có thể tạo ra một cột tạm thời trong bảng nơi tôi lưu trữ các chỉ số, nhưng tôi "tự hỏi nếu nó được sotred trong đối tượng hàng ở đâu đó.

Answer 1

Để truy cập vào các chỉ số trong trường hợp này, bạn truy cập vào name thuộc tính:

In [182]: 

df = pd.DataFrame([[1,2,3],[4,5,6]], columns=['a','b','c']) 
def rowFunc(row): 
    return row['a'] + row['b'] * row['c'] 

def rowIndex(row): 
    return row.name 
df['d'] = df.apply(rowFunc, axis=1) 
df['rowIndex'] = df.apply(rowIndex, axis=1) 
df 
Out[182]: 
    a b c d rowIndex 
0 1 2 3 7   0 
1 4 5 6 34   1 

Lưu ý rằng nếu điều này thực sự là những gì bạn đang cố gắng để làm điều đó các công trình sau đây và nhanh hơn nhiều:

In [198]: 

df['d'] = df['a'] + df['b'] * df['c'] 
df 
Out[198]: 
    a b c d 
0 1 2 3 7 
1 4 5 6 34 

In [199]: 

%timeit df['a'] + df['b'] * df['c'] 
%timeit df.apply(rowIndex, axis=1) 
10000 loops, best of 3: 163 µs per loop 
1000 loops, best of 3: 286 µs per loop 

Answer 2

apply() không phải là droid bạn đang tìm kiếm.

DataFrame.iterrows() cho phép bạn duyệt qua hàng, và truy cập vào tên của họ:

for name, row in df.iterrows(): 
    ...