如果在另一列中满足条件，则将列值附加到列表

KAs 发表于 Dev

谁

无法完全找到解决方案，这似乎很基本，但说我有：

df ={'ID' : [1, 1, 1, 1, 1, 1, 1, 2, 2],
     'x':[76.551, 77.529, 78.336,77, 76.02, 79.23, 77.733, 79.249,  76.077],
     'y': [151.933, 152.945, 153.970, 119.369, 120.615, 118.935, 119.115, 152.004, 153.027],
    'position': ['start', 'end', 'start', 'NA', 'NA','NA','end', 'start', 'end']}
df = pd.DataFrame(df)

df
   ID       x        y position
0   1  76.551  151.933    start
1   1  77.529  152.945      end
2   1  78.336  153.970    start
3   1  77.000  119.369       NA
4   1  76.020  120.615       NA
5   1  79.230  118.935       NA
6   1  77.733  119.115      end
7   2  79.249  152.004    start
8   2  76.077  153.027      end

我想用“开始”和“停止”点绘制唯一 ID 的路径。我想我可以这样做：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

cmap = cm.get_cmap('rainbow')
colors = cmap(np.linspace(0, 1, df['ID'].nunique()))
color_map = dict(zip(df['ID'].unique(), colors))

# fig, ax = plt.subplots()
plt.figure(figsize=(10,10))
for ID, subdf in df.groupby('ID'):
    plt.plot(subdf['x'], subdf['y'], marker=',', label=None, c=color_map[ID], linewidth=0.1)
# ax.legend()

plt.plot(x_start, y_start, 'og') # plot the start coords in green?
plt.plot(x_stop, y_stop, 'or') # plot the end coords in red?
plt.show()

但我无法弄清楚如何将开始和停止坐标附加到一个空数组......我认为这会起作用，但它没有：

x_start = []
x_end = []
y_start = []
y_end = []

for i in range(len(df)):
    if df.loc[df['position'] == 'start']:
        x_start.append(df['x'][i])
        y_end.append(df['y'][i])
    if df.loc[df['position'] == 'end']:
        x_end.append(df['x'][i])
        y_end.append(df['y'][i])

畜栏

您是否在寻找：

x_start, y_start = df.loc[df['position'] == 'start', ['x', 'y']].values.T.tolist()
x_end, y_end = df.loc[df['position'] == 'end', ['x', 'y']].values.T.tolist()

输出：

>>> x_start
[76.551, 78.336, 79.249]

>>> y_start
[151.933, 153.97, 152.004]

>>> x_end
[77.529, 77.733, 76.077]

>>> y_end
[152.945, 119.115, 153.027]

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。