如何超出列的最大值并返回列名以保存为变体

Question

我是一个非常初学者，热情的程序员想成为。我试图比较两列的csv，我从谷歌的趋势，并提取“赢家的词”到一个变体或列表。所以我可以和“趋势”中的其他关键词作比较。到目前为止，我设法：

csv的外观总是这样：

date,VR,metaverse
2004-01-01,17,0
2004-02-01,17,0
2004-03-01,18,0
2004-04-01,16,0
2004-05-01,17,0
2004-06-01,17,0

在：

csv1 = pd.read_csv ("search_trends.csv").drop("date", axis=1)
csv1 = pd.DataFrame(csv1)
result1 = csv1.max(axis=1)
result1.index = csv1.idxmax(axis=1)

退出：

VR    17
VR    17
VR    18
VR    16
VR    17
      ..
VR    45
VR    46
VR    45
VR    44
VR    40

Length: 226, dtype: int64
['VR', 'metaverse']

这里的诀窍是，我可能不知道列的名称，因此它必须是通用的。

Answer 1

还可以按索引访问列并检索其名称，但必须知道索引：

import pandas as pd

foo = pd.DataFrame()
foo['a'] = [1, 2, 3]

col_index = 0

max_of_column = max(foo.iloc[:, col_index])
name_of_column = foo.columns[col_index]