要避免使用循环和避免出现错误"a必须是一维的",可以使用np.random.choice函数的size参数来直接指定所需的数组形状。这样可以在一次函数调用中获得一个具有指定形状的数组。
以下是一个示例代码:
import numpy as np
# 创建一个包含10个随机整数的2x5数组
a = np.random.choice(10, size=(2, 5))
print(a)
输出:
[[7 4 2 8 6]
[2 7 3 4 9]]
在这个示例中,np.random.choice函数的第一个参数是随机数的范围(从0到9),size参数是一个元组,指定了所需的数组形状(2行5列)。这样就可以直接获得一个包含随机整数的2x5数组,而不需要使用循环或手动转换数组形状。
注意:在使用np.random.choice函数时,确保size参数指定的形状与所选随机数的总数相匹配,否则可能会出现"a必须是一维的"错误。