We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 82dbbba + 5560199 commit b4bb66fCopy full SHA for b4bb66f
powerplantmatching/export.py
@@ -73,7 +73,11 @@ def map_bus(df, buses):
73
"""
74
df = get_obj_if_Acc(df)
75
kdtree = KDTree(buses[["x", "y"]])
76
- buses_i = buses.index.append(pd.Index([np.nan]))
+ non_empty_buses = buses.dropna()
77
+ if non_empty_buses.empty:
78
+ buses_i = pd.Index([np.nan])
79
+ else:
80
+ buses_i = non_empty_buses.index.append(pd.Index([np.nan]))
81
return df.assign(bus=buses_i[kdtree.query(df[["lon", "lat"]].values)[1]])
82
83
0 commit comments