从熊猫数据帧创建 pd.Series 列表

Question

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

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

一些代码用 2 列显示我想要的输出，但我的用例如果有几十列，所以我不能手动编写它：

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    11    2Name: a, dtype: int64, 0    31    3Name: b, dtype: int64]或转换为to_dictwithto_dict(orient='series')并获取 的值dict，但顺序应更改为python under 3.6：desired_output = list(mydf[my_list_of_columns].to_dict(orient='series').values())

Answer 2

像这样的东西？import pandas as pdmydf = pd.DataFrame.from_dict({'a': {0: 1, 1: 2}, 'b': {0: 3, 1: 3}, 'c': {0: 10, 1: 3}})desired_output = []l = list(mydf)for i in l:    desired_output.append(list(eval('mydf.' + i)))print(desired_output)