在Sf中使用group_by()查找相距最远的两个点

import pandas as pd
import geopandas as gpd
from shapely.geometry import Point
from scipy.spatial.distance import pdist, squareform

# 加载数据
data = pd.read_csv('data.csv')

# 创建Sf数据结构
geometry = [Point(xy) for xy in zip(data['x'], data['y'])]
gdf = gpd.GeoDataFrame(data, geometry=geometry)

# 分组并计算距离
distances = gdf.groupby('group').agg(lambda x: pdist(x.geometry.values[:, None])).reset_index()

# 找到距离最远的两个点
max_distance = distances['distance'].max()
max_distance_indices = distances[distances['distance'] == max_distance].index

# 获取最远两个点的详细信息
point1 = gdf.loc[gdf['group'] == distances.loc[max_distance_indices[0], 'group']].iloc[0]
point2 = gdf.loc[gdf['group'] == distances.loc[max_distance_indices[0], 'group']].iloc[1]

# 输出结果
print("最远的两个点为：")
print(point1)
print(point2)
print("它们之间的距离为：", max_distance)