第17.18行,实在是不会改
import pandas as pd
data = pd.read_csv('Taxi_sz.csv')
groups = data.groupby('VehicleNum')
result = groups.apply(lambda x: (x['SLng'], x['SLat']))
result_list = [(lng.tolist()[:100], lat.tolist()[:100]) for lng, lat in result]
result_df = pd.DataFrame(result_list, columns=['SLng', 'SLat'])
result_df.index = result.index.tolist()
result_df.to_csv('new_data1.csv')
lat = result_df.loc[result_df.index == 22271]['SLat'].tolist()
lng = result_df.loc[result_df.index == 22271]['SLng'].tolist()
for i in range(len(lat[0])):
print('new BMap.Point('+str(lng[0][i])+','+str(lat[0][i])+'),')
可以使用 try-except 语句来处理这种情况。如下代码:
try:
lat = result_df.loc[result_df.index == 22271]['SLat'].tolist()
lng = result_df.loc[result_df.index == 22271]['SLng'].tolist()
for i in range(len(lat[0])):
print('new BMap.Point('+str(lng[0][i])+','+str(lat[0][i])+'),')
except IndexError:
print("索引超出范围")