如何根据坐标列表选择数据帧的某些值?

如果您将坐标转换为转置坐标的元组,则可以直接将其用作 ndarray 。您可以使用它将新数组中的值设置为 '' 或来自输入数据帧的相应值,取决于的值 inverted :

def select(df, coords, inverted=False):
    coords = tuple(np.array(coords).T)
    out = data = np.array(df)
    if inverted:
        out[coords] = ''
    else:
        out = np.full(data.shape, '')
        out[coords] = data[coords]
    return pd.DataFrame(out, columns=df.columns)

select(df, coords, True)
select(df, coords, False)

输出:

  col1 col2 col3
0    A    B    C
1    B    E    D
2         F    E
3    A    F
4    G    H    I

  col1 col2 col3
0
1
2    C
3              A
4

要从创建新的数据帧 df 和 coords 你可以做到:

out, coords = np.empty_like(df), np.array(coords)
out[:] = np.nan
out[coords[:, 0], coords[:, 1]] = df.to_numpy()[coords[:, 0], coords[:, 1]]

out = pd.DataFrame(data=out, copy=False, columns=df.columns).fillna("")
print(out)

打印:

  col1 col2 col3
0               
1               
2    C          
3              A
4               

相反:

out, coords = df.to_numpy(), np.array(coords)
out[coords[:, 0], coords[:, 1]] = ""

out = pd.DataFrame(data=out, copy=False, columns=df.columns)
print(out)

打印:

  col1 col2 col3
0    A    B    C
1    B    E    D
2         F    E
3    A    F     
4    G    H    I