python - 从 panda DataFrame 创建 pd.Series 列表

Question

我有一个包含很多列的 DataFrame，以及我想要传递给算法的列名称列表。该算法要求我传入的参数不是数据帧，而是一个列表，其中每个元素都是我的 pd.Series 列之一。

我认为这个问题可能已经得到解答，但我找不到...是否有函数或方法来实现此转换data.Frame => pd.Series 列表？

一些代码显示我想要的两列输出，但我的用例如果有几十列，所以我无法手动编写:

mydf = pd.DataFrame.from_dict({'a': {0: 1, 1: 2}, 'b': {0: 3, 1: 3}, 'c': {0: 10, 1: 3}})
my_list_of_columns=["a", "b"]
desired_output = [mydf.a, mydf.b]

感谢您的帮助

Answer 1

使用列表理解:

desired_output = [mydf[x] for x in my_list_of_columns]
print (desired_output)
[0    1
1    2
Name: a, dtype: int64, 0    3
1    3
Name: b, dtype: int64]

或转换为to_dict使用 to_dict(orient='series') 并获取 dict 的值，但在 3.6 下的 python 中应更改排序:

desired_output = list(mydf[my_list_of_columns].to_dict(orient='series').values())