python - 如何检查数组是否有 2 列或 3 列？

Question

如何检查给定数组是 2 列还是 3 列？以及如何在找到后将第三列值设置为零？

我正在尝试调试程序，我知道如何找到数组的形状，但让我们假设这是数组。

array([[   0., -300.,   25.],
       [  25.,    2.,   25.],
       [  50.,    6.,   25.],
       [  75.,    2.,   25.],
       [  85.,    2.,   25.],
       [  30.,    2.,   25.],
       [  10.,    4.,   23.]])

多谢。

标签： pythonnumpy

Answer 1

arr = np.array([[   0., -300.,   25.],
       [  25.,    2.,   25.],
       [  50.,    6.,   25.],
       [  75.,    2.,   25.],
       [  85.,    2.,   25.],
       [  30.,    2.,   25.],
       [  10.,    4.,   23.]])

if arr.shape[1] == 3:
    arr[:, 2] = 0