python - 在 Python 中处理基本数组的首选方法是什么？

Question

在使用 Matlab 多年后，我正在学习 python。我遇到了障碍，在谷歌搜索中没有找到太多帮助。

我想知道处理基本数组的最佳方法。我应该使用 numpy、scipy、array、numarray 还是其他东西？

例如，采用以下 Matlab 代码。

a = rand(10,1)
b = rand(10,1)


c = b > 0.5
d = a .* b
e = a + b
f = mean(a)
g = sum(b)

将其转换为 python 的最佳方法是什么？

Answer 1

如果您要使用数字数组进行数学运算，那么您绝对应该使用 NumPy；甚至还有一个migration guide面向 MATLAB 用户。

NumPy 执行许多与 MATLAB 相同的数组广播操作，因此使用起来应该非常自然。您的代码可以写为:

import numpy as np
a = np.random.rand(10,1)
b = np.random.rand(10,1)
c = b > 0.5
d = a * b # Note that * in NumPy is always elementwise (use .dot for matrix multiplication)
e = a + b
f = a.mean() # Can also use np.mean(a)
g = b.sum() # Can also use np.sum(b)