python - 具有逻辑索引的散点图

Question

我有一个 100x2 数组 D 和一个 100x1 数组 c(条目 +/- 1)，我正在尝试绘制 D 中与 c = 1 相对应的列的散点图。

我尝试了这样的方法:plt.scatter(D[0][c==1],D[1][c==1])但它抛出了IndexError :数组索引过多

我知道我使用了列表理解或类似的东西。我对 Python 相当陌生，因此在格式上遇到了困难。

非常感谢。

Answer 1

概念

您可以使用 np.where 仅选择 D 中数组 C 中 1 的行:

D = np.array([[0.25, 0.25], [0.75, 0.75]])
C = np.array([1, 0])

使用np.where，我们可以仅选择C中1的行:

>>> D[np.where(C==1)]
array([[0.25, 0.25]])

示例根据您的实际数据:

D = np.random.randn(100, 2)
C = np.random.randint(0, 2, (100, 1))

valid = D[np.where(C.ravel()==1)]

import matplotlib.pyplot as plt
plt.scatter(valid[:, 0], valid[:, 1])

输出: