将索引列表转换为二维numpy数组的最快方法

import numpy as np

# 假设索引列表为index_list
index_list = [0, 1, 2, 3, 4, 5]

# 将索引列表转换为numpy数组
index_array = np.array(index_list)

# 将一维索引数组转换为二维numpy数组
# 这里假设转换为2行3列的二维数组
two_dim_array = index_array.reshape(2, 3)

# 将数组的数据类型转换为int
two_dim_array = two_dim_array.astype(int)

# 打印结果
print(two_dim_array)

[[0 1 2]
 [3 4 5]]