主页 / 问题 / 1114774
Accepted
Oleg Chaika
Asked:2020-04-24 02:06:31 +0000 UTC2020-04-24 02:06:31 +0000 UTC

比较 2D Python 矩阵的元素

  • 772

下一个问题：如何在带有 NumPy 的 Python 中遍历二维矩阵的每个元素，并将其与某个值进行比较？我的代码如下所示：

print("Укажите размер массива, столбцов:")
n = int(input())
print("Укажите размер массива, строк:")
m = int(input())
universe = np.random.randint(2, size=(m, n))
print (universe)
for row in universe:
      for x in row:
          if universe[[row], [x]] == 1:
              j += 1
print (j)

就我而言，它给出了一个错误：

if universe[[row], [x]] == 1:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

请告诉我正确的方法，谢谢。

python

2 个回答

  1. Best Answer

    代替

    for row in universe:
      for x in row:
          if universe[[row], [x]] == 1:

    申请

    for row in range(universe.shape[0]):
    for x in range(universe.shape[1]):
        if universe[row, x] == 1:
    • 2

  2. 你可以这样比较：

    universe == 1

    结果将是一个大小相同的数组，其中包含指示条件是否为真的 bool 值。

    您可以像这样计算条目的数量：

    number_of_ones = np.count_nonzero(universe == 1)
    • 0

相关问题