在熊猫数据框中创建索引第一行

Question

我想知道是否可以根据索引创建每个组的第一行，即该索引的名称。假设我们有一个这样的df：

dic = {'index_col': ['a','a','a','b','b','b'],'col1': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(dic).set_index('index_col')

是否可以将上面的数据框转换为类似于下面的数据框？这里发生的事情是索引已被重置，并且对于每个组，第一行是索引名称？

Answer 1

结果是pandas.Series；

df_list = []
for label , group in df.groupby('index_col'):
    df_list.append(pandas.concat([pandas.Series([label]), group['col1']]))

df_result = pandas.concat(df_list).reset_index(drop=True)

输出；

0    a
1    1
2    2
3    3
4    b
5    4
6    5
7    6
dtype: object

如果您需要数据框，请致电 df_result.to_frame()。