python - 将列值分配给数据框中的变量

Question

我将 Excel 工作表读入数据框中。是否可以循环遍历列并将包含每列值的列表分配给以列名作为变量名的变量？作为一个简单的例子，我有数据框

val = [[1,396,29],[2,397,29],[3,395,29],[4,393,29],[5,390,29],[6,398,29]]
df=pd.DataFrame(val,columns=['Hours','T1T_in','T1p_in'])
df

Hours   T1T_in  T1p_in
0   1   396 29
1   2   397 29
2   3   395 29
3   4   393 29
4   5   390 29
5   6   398 29

所以我想要一个循环创建以列名称作为变量的列表？

Hours = [1,2,3,4,5,6]
T1T_in = [396,397,395,393,390,398]
T1p_in = [29,29,29,29,29,29,29]

我找到了一种获取名称但无法分配值的解决方案。谢谢您的帮助

Answer 1

获取列元素作为列表的最简单方法是使用pandas.Series.tolist():

Hours = df.Hours.to_list()
T1T_in = df.T1p_in.to_list()
T1p_in = df.T1p_in.to_list()

如果需要，您还可以使用 for 循环(如您所说)来获取列和行:

data = {}
for column_name, rows in df.iteritems():
    data[column_name] = rows.to_list()

print(data)

输出:

{'Hours': [1, 2, 3, 4, 5, 6],
 'T1T_in': [396, 397, 395, 393, 390, 398],
 'T1p_in': [29, 29, 29, 29, 29, 29]}

上述结果也可以通过以下方式实现:

df.to_dict('list')

正如@anky_91所说