跳转至

如何使用 NumPy 移除只包含 0 的数组行?

原文:https://www . geeksforgeeks . org/如何使用-numpy/ 移除仅包含-0 的数组行

Numpy 库提供了一个名为 numpy.all() 的函数,当传递给第一个参数的 n-d 数组的所有元素都为真时返回真,否则返回假。因此,要确定包含 0 的整行可以通过指定 axis=1 来移除。它将遍历每一行,并检查第一个参数中给出的条件。

示例:

data=[[1,2,3]
      [0,0,0]
      [9,8,7]]
After removing row with all zeroes:
data=[[1,2,3]
      [9,8,7]]

例 1:

遵循的方法:

  • 取一个 numpy n-d 数组。
  • 使用 numpy.all()函数删除只包含零的行。
  • 打印 n-d 数组。

蟒蛇 3

import numpy as np
# take data
data = np.array([[1, 2, 3], [0, 0, 0], [4, 5, 6],
                 [0, 0, 0], [7, 8, 9], [0, 0, 0]])
# print original data having rows with all zeroes
print("Original Dataset")
print(data)

# remove rows having all zeroes
data = data[~np.all(data == 0, axis=1)]

# data after removing rows having all zeroes
print("After removing rows")
print(data)

输出:

例 2:

遵循的方法:

  • 使用 numpy.random.choice()方法,取 0-10 之间的 20 个随机数。
  • 使用 resform()方法将它们按行和列对齐。
  • 显式地将一些行标记为完全 0。
  • 删除全为零的行。
  • 打印数据集。

蟒蛇 3

import numpy as np
# take random data

# random.choice(x,y) will pick y elements from range (0,(x-1))
data = np.random.choice(10, 20)

# specify the dimensions of data i.e (rows,columns)
data = data.reshape(5, 4)

# print original data having rows with all zeroes
print("Original Dataset")
print(data)

# make some rows entirely zero
data[1, :] = 0  # making 2nd row entirely 0
data[4, :] = 0  # making last row entirely 0

# after making 2nd and 5th row as 0
print("After making some rows as entirely 0")
print(data)
data = data[~np.all(data == 0, axis=1)]

# data after removing rows having all zeroes
print("After removing rows")
print(data)

输出:



回到顶部