熊猫使用特定的字符串和数组/列表选择数据框列

时间:2020-09-10 00:57:20

标签: pandas

我有一个包含数百列(库存)的数据框。我的问题是,我需要始终拉一个特定的列(日期),然后再拉一个其他的数组/列表(动态)。

以前我是在做这样的事情:

df = stocks[['date', 'AAPL', 'AMZN']]

但是,如果现在我需要根据某个行业动态选择股票,我不确定如何使它们一起发挥作用。我只能在不使用日期的情况下提取列表:

print(rowData['symbol'])
3    [APA.OQ, BKR.N, COG.N, CVX.N, CXO.N, COP.N, DV...
Name: symbol, dtype: object

selection = rowData['symbol'].explode()
df = stocks[selection]

我还如何获取日期值?像这样的东西行不通:

df = stocks[['date'][selection]]

谢谢

1 个答案:

答案 0 :(得分:0)

让我们尝试

df = stocks[['date'] + rowData['symbol'].iloc[0]]