通過熊貓數據幀循環訪問函數

Question

我的目標是轉換一個數據幀中的數據並將結果輸出到新的數據幀。這裏是我到目前爲止，使用簡化的數據幀：

import math 
import pandas as pd 
data = {'A':[1,4,3,5,7],'B':[0,6,3,0,2],'C':[1,1,3,0,4]} #sample data 
df = pd.DataFrame(data) 
transDF = pd.DataFrame() #empty dataframe for results 

def Chord(y): #Chord transformation function 
    ySUM = sum(a*a for a in y) 
    ySUMsqrt = math.sqrt(ySUM) 
    yPRIME = [] 
    for a in y: 
     RESULT = a/ySUMsqrt 
     yPRIME.append(RESULT) 
    return yPRIME 

for Yi, row in df.iterrows(): #my attempt at a loop 
    Yrow = df.loc[df.index == Yi] 
    y = yRow.values.tolist() 
    tfRow = float(Chord(y)) 
    transDF = transDF.append(tfRow) 

，如果我只是給它一個列表本身工作的功能，但是當我嘗試環路我得到，說：「不能乘序列錯誤類型'list'的非整型。我嘗試過儘可能多的修改我的循環，但在這一點上，我已經沒有想法了。我將不勝感激任何見解！

Answer 1

IIUC，我不認爲需要這個問題iterrows。

import math 
data = {'A':[1,4,3,5,7],'B':[0,6,3,0,2],'C':[1,1,3,0,4]} #sample data 
df = pd.DataFrame(data) 
transDF = pd.DataFrame() #empty dataframe for results 

def Chord(y): #Chord transformation function 
    ySUM = sum(a*a for a in y) 
    ySUMsqrt = math.sqrt(ySUM) 
    yPRIME = [] 
    for a in y: 
     RESULT = a/ySUMsqrt 
     yPRIME.append(RESULT) 
    return yPRIME 

transDF = df.apply(Chord) 
print(transDF) 

輸出：

 A   B  C 
0 0.1 0.000000 0.19245 
1 0.4 0.857143 0.19245 
2 0.3 0.428571 0.57735 
3 0.5 0.000000 0.00000 
4 0.7 0.285714 0.76980 

Answer 2

你的代碼是非常低效的。 在熊貓中循環播放幾乎總是不必要的，循環播放單個元素應該更加罕見。

利用numpys矢量化！

import pandas as pd 
import numpy as np 

def chord_transform(row): 
    return row/np.sum(row**2) 

data = {'A':[1,4,3,5,7],'B':[0,6,3,0,2],'C':[1,1,3,0,4]} #sample data 
df = pd.DataFrame(data) 
df_chord = df.apply(chord_transform, axis=1)