python - 将嵌套列表转换为 numpy 数组

Question

这个问题已经有答案了:

List of lists into numpy array (9 个回答)

已关闭 5 年前。

我有一些数据，我已将其排列为 [[[0, 1], [2, 3]], [[4, 5], [6,7]]] 的形式，例如3个维度。但是，我正在使用的函数需要 numpy数组才能工作，所以我需要将其从嵌套列表转换为一个。每当我尝试时，我总是发现前两层列表会转换，但最后一层不会转换？例如如果我尝试打印:

print(data)          # Prints a numpy.ndarray
print(data[0][0])    # Prints a numpy.ndarray
print(data[0][0][0]) # Prints a list

我尝试了几种不同的方法来转换它，但它们都返回这样的数据？有人知道我能做什么吗？

Answer 1

对您的数据所指的内容有点困惑，但给出

import numpy as np

x = [[[0, 1], [2, 3]], [[4, 5], [6,7]]]

# convert to numpy.ndarray
y = np.asarray(x)
print(type(y)) # y is now a numpy.ndarray

# Checking dimensions
print(y.shape)  # (2,2,2) - 3D array