连接具有相同 ID Pandas DataFrame 的列

Question

我有一个名为 weather 的 DataFrame，其结构如下：

    STATION     DATE        ELEM    VALUE
0   US1MNCV0008 20170101    PRCP    0
1   US1MNCV0008 20170101    SNOW    0
2   US1MISW0005 20170101    PRCP    0
3   US1MISW0005 20170101    SNOW    0
4   US1MISW0005 20170101    SNWD    0

我想使用日期和电台组合行以获得以下内容：

    STATION     DATE        ELEM  VALUE ELEM  VALUE ELEM VALUE
0   US1MNCV0008 20170101    PRCP  0     SNOW  0
1   US1MISW0005 20170101    PRCP  0     SNOW  0     SNWD    0

我正在尝试通过使用：

weather.groupby(['station', as_index=False).agg(lambda x: x.tolist())

但这会创建列表，这不是我想要的。如何进行聚合？

Answer 1

你可以使用：

df = (df.set_index(['STATION','DATE', df.groupby(['STATION','DATE']).cumcount()])
        .unstack()
        .sort_index(axis=1, level=1))
df.columns = ['{}_{}'.format(i, j) for i, j in df.columns]
df = df.reset_index()
print (df)
       STATION      DATE ELEM_0  VALUE_0 ELEM_1  VALUE_1 ELEM_2  VALUE_2
0  US1MISW0005  20170101   PRCP      0.0   SNOW      0.0   SNWD      0.0
1  US1MNCV0008  20170101   PRCP      0.0   SNOW      0.0    NaN      NaN

解释：

1. 通过STATION 和DATE 和cumcount 获取每个组的计数
2. 由set_index 创建MultiIndex
3. 由unstack重塑
4. 在列中展平MultiIndex
5. 将index 转换为reset_index 的列

---

或者使用GroupBy.apply为每个组创建DaatFrame，最后解决方法同上：

df = (df.groupby(['STATION','DATE'])['ELEM','VALUE']
       .apply(lambda x: pd.DataFrame(x.values, columns=x.columns))
       .unstack()
       .sort_index(axis=1, level=1))
df.columns = ['{}_{}'.format(i, j) for i, j in df.columns]
df = df.reset_index()
print (df)
       STATION      DATE ELEM_0 VALUE_0 ELEM_1 VALUE_1 ELEM_2 VALUE_2
0  US1MISW0005  20170101   PRCP       0   SNOW       0   SNWD       0
1  US1MNCV0008  20170101   PRCP       0   SNOW       0    NaN     NaN