如何基于数据框中的列名创建变量？

Question

我想根据数据框的列名在python中创建变量。不知道这是否可行，因为我是Python的新手。可以说我的df看起来像这样：

ID  Date         Name    Counts
1   04/04/2018   JUOM    150
2   05/04/2018   1XMK    22
3   05/04/2018   N2IS    130

如何创建一个变量，可以在其中引用此数据框中的“日期”和“名称”列？

Answer 1

如果要将变量x分配给列Date和Name，则可以使用df[['col1','col2',...]]语法对数据帧进行子集化。外部方括号是pandas数据框对象的索引器操作，而内部方括号创建所需列名称的列表。

x= df[['Date','Name']]

哪个返回：

         Date  Name
0  04/04/2018  JUOM
1  05/04/2018  1XMK
2  05/04/2018  N2IS

如果您不想将数据包含在数据框中，则可以使用.values属性来获取数据。

x= df[['Date','Name']].values

array([['04/04/2018', 'JUOM'],
       ['05/04/2018', '1XMK'],
       ['05/04/2018', 'N2IS']], dtype=object)

如果您使用的熊猫版本为0.24.0，他们建议将.values替换为to_numpy()。