python - 参数化 Pandas 中的列名

Question

我想知道是否可以参数化列名。我有一个例子

import pandas as pd
df = pd.DataFrame({'region':[1,2,3,4,5,6],
               'id_1':[456, 456,    342,    987,    424,    424],
               'id_2':[98,  87, 65, 65, 32, 43],
               'id_3':[12,  12, 34, 34, 56, 56],
               'count':[1,0,1,1,0,0]})

reg_id=df.groupby(['region', 'id_1'])['count'].value_counts().unstack(fill_value=0)

这个想法是为 id_columns 的参数化引入一条线。像这样的东西：

**列=id_1

** reg_id=df.groupby(['region', column ])['count'].value_counts().unstack(fill_value=0)

更改此参数，我将能够按我需要的列进行分组，而无需更改脚本中的任何内容。现在我为每列“id”有几个脚本。

标签： pythonpandas