python - 用于数组索引的 Matlab 逻辑 - Python 等效项

Question

在 MATLAB 中，有一种掩码类型的运算，其逻辑如下所示:

A=randn(1,10)  % vector with 10 random values

% logical mask indexing:
A(A<0)=NaN     % setting values smaller than 0 to NaN

Python中有这么简单的等效操作吗？我没有找到。

Answer 1

import random
a = [random.random() - 0.5 for _ in range (10)]
a = [None if i < 0 else i for i in a]

如果您使用长数组(并且正在执行类似 MATLAB 的操作)，您可能需要考虑 numpy，因为它的效率更高。

import numpy as np
a = np.random.rand(10) - 0.5
a[a<0] = np.NaN