python - 为 np.array(1.) 指定默认数据类型

Question

有没有办法指定与 np.array(1.) 等结构一起使用的默认数据类型？

特别是我希望 np.array(1.) 是 np.float32 和 np.array(1) 是 np.int32。相反，我得到了 np.float64 和 np.int64

Answer 1

默认值取决于您的系统。在 64 位系统上，默认类型为 64 位。在 32 位系统上，默认类型为 32 位。无法更改使用不同的系统 C header 重新编译 numpy 的默认值。

您当然可以明确指定数据类型，例如

>>> x = np.array(1, dtype='int32')

编辑:正如 kazemakase 在下面提到的，以上仅适用于 int32/int64。在最近的 numpy 版本中，无论系统如何， float 的默认值都是 float64。