Pandas选择多列数据的最佳做法

Question

我目前有一个非常冗余的代码来处理来自pandas数据帧的行的多个列。看起来像这样：

info = []
info.append(row['column name'])
...
(multiple lines of mere appending)

我将此列表传递给另一个python脚本，该脚本填充网页上返回的值。我必须改进这个以摆脱冗余，我不知道该怎么做。

执行此类任务的最佳做法是什么？

Answer 1

您可以传递列名列表。

import pandas as pd

df = pd.DataFrame({'x':[1, 2, 3], 'y':[5, 6, 7], 'z':['a', 'b', 'c']})

选择包含iloc的行，并使用名称列表选择列，例如如果你只想要'x'和'z'：

df.iloc[1][['x', 'z']]

返回：

x    2
z    b
Name: 1, dtype: object