熊猫数据框评估中的动态列名称

Question

我按如下方式引用数据框（Sales是列名称）：

total = pd.to_numeric(sales_df.Sales.str.replace("$", "")).sum()

但是我不希望Sales被硬编码，我想要一个变量来使它变得动态。这是怎么做到的？

TIA

Answer 1

您可以使用方括号对列进行索引：

self.view

因此，当您接受输入为df['col_name'] 时，您可以这样做：

str

另外，将列作为属性访问可能会导致模糊行为。例如，有一个名为total = pd.to_numeric(sales_df[user_input_name].str.replace("$", "")).sum() 的列，并且您尝试index可能具有与列df.index不同的值，或者您有一个名称与任何有效的df方法相同的列，如{ {1}}或df['index']这会导致语法错误。

所以我强烈建议您使用方括号来选择列。