python - numpy 中 ogrid 的 dtype 是否可以指定？

Question

在Python的numpy中，为什么ogrid总是产生int64结果？

对于我的应用程序，由于内存限制(当输出组件稍后一起广播时会发挥作用)，我不想使用 int64。有没有比事后重铸更好的选择:

y, x = np.ogrid[:9000,:9000]
y = y.astype(np.int16)
x = x.astype(np.int16)

对于大多数其他 numpy 调用，更简洁的解决方案是使用 dtype=... 可选参数，但 ogrid 不会作为函数调用。相反，它似乎与像 a+b 这样的运算符相当，只不过它们通常有像 np.add(a,b,dtype=np.int8) 这样的替代选项。

Answer 1

您可以使用ix_生成相同的形状，并完全控制数据类型:

In [476]: np.ix_(np.arange(5,dtype=float),np.arange(5,dtype=np.int16))
Out[476]: 
(array([[ 0.],
        [ 1.],
        [ 2.],
        [ 3.],
        [ 4.]]), array([[0, 1, 2, 3, 4]], dtype=int16))
In [477]: np.ogrid[:5,:5]
Out[477]: 
[array([[0],
        [1],
        [2],
        [3],
        [4]]), array([[0, 1, 2, 3, 4]])]

meshgrid 以及:

In [488]: np.meshgrid(np.arange(5, dtype=float), np.arange(5, dtype=np.int16), sparse=True, indexing='ij')
Out[488]: 
[array([[ 0.],
        [ 1.],
        [ 2.],
        [ 3.],
        [ 4.]]), array([[0, 1, 2, 3, 4]], dtype=int16)]