Python Pandas Groupby列表列表

Question

我是Python的新手，正在尝试将在两个对我有用的单独程序中创建的功能组合在一起。

目标是按各种说明对值进行分组，然后按日期对数据集的平均值进行分组。 我已经使用Pandas Groupby成功地做到了这一点。

我要评估的描述之一是在数据集中每个点的给定距离内求平均值。 到目前为止，我一直使用邮政编码作为位置描述来对此进行近似。 另外，我已经能够使用Geopy使用GPS点确定数据集中所需距离内的所有其他点。 这为我提供了所需距离内数据集中每个ID的ID列表。

这是一个示例数据集：

ID  Date    Value   Color  Location
1    1      1234    Red    60941
1    2      51461   Red    60941
1    3      6512    Red    60941
1    4      5123    Red    60941
1    5      48413   Red    60941
2    1      5416    Blue   60941
2    2      32      Blue   60941
2    3      18941   Blue   60941
2    4      5135    Blue   60941
2    5      1238    Blue   60941
3    1      651651  Blue   60450
3    2      1777    Blue   60450
3    3      1651    Blue   60450
3    4      1968    Blue   60450
3    5      846     Blue   60450
4    1      1689    Red    60941
4    2      1651    Red    60941
4    3      184     Red    60941
4    4      19813   Red    60941
4    5      132     Red    60941
5    1      354     Yellow 60450
5    2      684     Yellow 60450
5    3      489     Yellow 60450
5    4      354     Yellow 60450
5    5      846     Yellow 60450

这是我目前使用邮政编码位置描述进行的Pandas代码：

average_df = data_df['Value'].groupby([data_df['Location'],data_df['Color'],data_df['Date']]).mean()

有没有办法将从Geopy获得的列表传递到Groupby来代替我目前拥有的['Location']组？ 例如，Groupby List（ID）[列表1：（1,2,3），列表2：（3,1,5），列表3：（2,3,4）]然后是颜色和日期。

我浏览了Pandas文档并搜索了该网站，但在Pandas Groupby中找不到任何使用列表的人，因此我不确定是否有可能。 也许我需要在一个numpy数组中执行此操作？ 对于任何反馈，我们都表示感谢。

Answer 1

熊猫将轻松按布尔列表进行分组。 因此，您所需要做的就是获取每行是否在附近的列表。 最简单的方法是使用列表理解：

df = pandas.DataFrame({'value': [3,2,3,6,4,1], 'location': ['a', 'a', 'b', 'c', 'c', 'c']})
nearby_locations = ['a','b']
is_nearby = [(loc in nearby_locations) for loc in df['location']]  
# is_nearby = [True, True, True, False, False, False]
df.groupby(is_nearby).mean()

这将输出：

          value
False  3.666667
True   2.666667