python - 按 pandas 数据框中的字段分组

Question

我有一个包含以下字段的数据框。对于每个 Id，我有两条记录，分别代表不同的纬度和经度。我正在尝试实现一个结果数据帧，该数据帧根据 id 按当前数据帧进行分组，并将其纬度和经度放入不同的字段中。

我尝试过按功能分组，但没有得到预期的结果。任何帮助将不胜感激。

Id  StartTime   StopTime    Latitude    Longitude
101 14:42:28    14:47:56    53.51       118.12
101 22:10:01    22:12:49    33.32       333.11

结果:

Id  StartLat    StartLong   DestLat DestLong
101 53.51       118.12      33.32       333.11

Answer 1

您可以使用groupby使用应用函数将 DataFrame 返回到 Series:

df = df.groupby('Id')['Latitude','Longitude'].apply(lambda x: pd.Series(x.values.ravel()))
df.columns = ['StartLat', 'StartLong', 'DestLat', 'DestLong']
df = df.reset_index()
print (df)
    Id  StartLat  StartLong  DestLat  DestLong
0  101     53.51     118.12    33.32    333.11

如果出现问题:

TypeError: Series.name must be a hashable type

尝试将Series更改为DataFrame，但随后需要unstack与 droplevel :

df = df.groupby('Id')['Latitude','Longitude']
       .apply(lambda x: pd.DataFrame(x.values.ravel()))
       .unstack()
df.columns = df.columns.droplevel(0)
df.columns = ['StartLat', 'StartLong', 'DestLat', 'DestLong']
df = df.reset_index()
print (df)
    Id  StartLat  StartLong  DestLat  DestLong
0  101     53.51     118.12    33.32    333.11