Question

我有一个数组A=[A0,A1]，其中A0 is a 4x3 matrix, A1 is a 3x2 matrix。我想比较一个浮点数，比如1.0，元素。预期的返回B=(A>1.0)是一个与A大小相同的数组。如何实现这一目标？

我可以将A复制到C，然后将C中的所有元素重置为1.0，然后进行比较，但我认为python（numpy / scipy）必须有更聪明的方法来做到这一点...... 谢谢。

Answer 1

假设我们提到的数组阵列具有相同的形状：

>>> A=np.array([np.random.random((4,3)), np.random.random((3,2))])
>>> A
array([ array([[ 0.20621572,  0.83799579,  0.11064094],
       [ 0.43473089,  0.68767982,  0.36339786],
       [ 0.91399729,  0.1408565 ,  0.76830952],
       [ 0.17096626,  0.49473758,  0.158627  ]]),
       array([[ 0.95823229,  0.75178047],
       [ 0.25873872,  0.67465796],
       [ 0.83685788,  0.21377079]])], dtype=object)

我们可以使用where子句测试每个元素：

>>> A[0]>.2
array([[ True,  True, False],
       [ True,  True,  True],
       [ True, False,  True],
       [False,  True, False]], dtype=bool)

但不是整件事：

>>> A>.2
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

因此只需重建阵列B：

>>> B=np.array([a>.2 for a in A])
>>> B
array([ array([[ True,  True, False],
       [ True,  True,  True],
       [ True, False,  True],
       [False,  True, False]], dtype=bool),
       array([[ True,  True],
       [ True,  True],
       [ True,  True]], dtype=bool)], dtype=object)

Answer 2

对1个矩阵使用列表推导

def compare(matrix,flo):
    return  [[x>flo for x in y] for y in matrix]

假设我正确理解你的问题，例如

matrix= [[0,1],[2,3]]
print(compare(matrix,1.5))

应打印[[False, False], [True, True]]

有关矩阵列表：

def compareList(listofmatrices,flo):
    return [[[x>flo for x in y] for y in matrix] for matrix in listofmatrices]

或

def compareList(listofmatrices,flo):
    return [compare(matrix,flo) for matrix in listofmatrices]

更新：递归函数：

def compareList(listofmatrices,flo):
    if(isinstance(listofmatrices, (int, float))):
        return listofmatrices > flo
    return [compareList(matrix,flo) for matrix in listofmatrices]

Python：将数组元素与float进行比较

2 个答案: