我正在寻找 numpy 数组中至少有一个单元格为 1 的列。
Input Array
[0,0,1,0,0,0,1,0,0,1]
[0,1,0,0,0,0,0,0,1,0]
[0,0,0,0,0,0,0,1,0,0]
[0,0,0,1,0,0,1,0,0,0]
Expected Output
[0,1,1,1,0,0,1,1,1,1]
最佳答案
使用numpy.any使用 axis=0(沿第一个轴展平，即沿行展平):
>>> np.any(a, axis=0)
array([False, True, True, True, False, False, True, True, True, True], dtype=bool)
当然，您可以轻松地将 bool 数组转换为整数:
>>> np.any(a, axis=0)*1
array([0, 1, 1, 1, 0, 0, 1, 1, 1, 1])
关于python - Numpy array ，查找带有 1s 的列，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/27517658/