python - 在数据框中连接多个 for 循环的更好方法

Question

所以我有一个包含很多列的数据框，并且我正在运行多个 for 循环来创建要在我想要的函数中使用的变量。是否有更好的（串联格式）方式/格式来运行这些循环？

for x in df['A']:
    L = x

for y in df['B']:
    M = y

for w in df['C']:
    N = w

for v in df['D']:
    O = v

也可能同时提高执行速度。

标签： pythonpandasdataframe

Answer 1

您可以通过按列表查找列来创建 numpy 数组，并且DataFrame.to_numpy：

for L,M,N,O in df[['A','B','C','D']].to_numpy():
    print (L, M, N, O)
     

或使用DataFrame.itertuples：

for row in df.itertuples():
    (L, M, N, O) = (row.A,row.B,row.C,row.D)