NumPy数组的原地类型转换

20 浏览
0 Comments

NumPy数组的原地类型转换

给定一个int32的NumPy数组,如何将其原地转换为float32?所以基本上,我想做的是a = a.astype(numpy.float32),而不复制数组。这个数组很大。

这样做的原因是我有两个计算a的算法。其中一个返回一个int32的数组,另一个返回一个float32的数组(这是两个不同算法固有的特性)。所有后续的计算都假设a是一个float32的数组。

目前我在一个通过ctypes调用的C函数中进行转换。有没有办法在Python中做到这一点?

0