如何在pandas系列中使用用户定义的输入作为列名

Question

我希望了解如何在列名中使用用户定义的变量。我正在使用熊猫。我有一个数据框，其中有几列具有相同的格式，但代码将针对不同的列名运行。每次只有名字的第一部分实际发生变化时，我不想要输入不同的列名。

例如，

df['input_same_same']

代码将调用不同的列，其中只有列的第一部分不同，其余列保持不变。

是否可以采取以下措施：

vari='cats' (and the next time I run I can input dogs, pigs, etc)

代表

df['vari_count_litter']

我尝试在列名中使用％s，但这并不起作用。

我很感激任何见解或理解这是如何可行的。谢谢！

Answer 1

如果我理解正确，你可以df[vari+'_count_litter']。但是，您可能最好使用可以让df[vari, 'count_litter']执行的MultiIndex。很难说如何设置它而不知道你的数据结构是什么以及你想如何访问它。