python ones生成的数组类型转换

Python ones 函数生成的数组类型是 numpy.ndarray。可以使用 numpy 中的 astype() 方法将其转换为其他数据类型。例如，将其转换为整数类型可以使用以下代码：

import numpy as np

arr = np.ones((3,3))
int_arr = arr.astype(int)

print(int_arr)

输出结果为：

[[1 1 1]
 [1 1 1]
 [1 1 1]]