numpy.random.randint返回的数据类型及数组格式适配问题

1. np.random.randint返回的数据类型 #

执行p = np.random.randint(0, 10, size=(15, 2))后，p是numpy ndarray（多维数组）——这是numpy库的核心数据结构。你可以直接用type(p)验证，输出会是<class 'numpy.ndarray'>。

2. 让普通嵌套列表支持numpy式切片 #

你用append([int, int1])生成的是普通Python嵌套列表，这种结构不支持[:,0]这类numpy风格的切片语法。要实现相同的调用效果，只需要把普通列表转换成numpy数组就行：

示例代码：

import numpy as np

# 你的原有列表生成逻辑
my_list = []
my_list.append([10, 10])
my_list.append([10, 22])
# 继续添加更多元素...

# 转换为numpy数组
p = np.array(my_list)

# 现在就能用切片调用了
print(p[:, 0])

转换后的p和np.random.randint生成的结构完全一致，都是numpy ndarray，支持所有numpy的切片、索引操作。

内容的提问来源于stack exchange，提问作者SocketLearner1302