使用NumPy创建数组时，如何保留自定义非命名复合dtype？

当你使用np.dtype((np.float32, 3))这种形式创建数组时，NumPy会自动将这种子数组类型“展开”为更高维度的数组，将dtype降级为基础的float32，同时扩展shape，这是NumPy的默认行为。要保留你定义的原始dtype且不需要字段名称，可以通过以下两种方式实现：

方法1：使用空名称的结构化dtype #

定义一个没有字段名的结构化dtype，NumPy会将其视为复合类型，不会自动展开：

import numpy as np

# 定义空名称的结构化dtype
dt = np.dtype([("", np.float32, 3)])
Z = np.zeros(10, dt)

# 验证结果
print(Z.dtype)  # 输出 dtype([('', '<f4', (3,))])
print(Z.shape)  # 输出 (10,)

# 访问数值数组（得到(10,3)的float32数组）
print(Z[''].shape)  # 输出 (10, 3)

方法2：直接构造ndarray对象 #

通过np.ndarray直接指定shape和dtype，绕开NumPy的自动展开逻辑（注意：此方式创建的数组内容未初始化，需手动填充或后续赋值）：

import numpy as np

dt = np.dtype((np.float32, 3))
# 创建未初始化的数组，shape为(10,)，dtype保留原始定义
Z = np.ndarray(shape=(10,), dtype=dt)

# 验证结果
print(Z.dtype)  # 输出 dtype(('float32', (3,)))
print(Z.shape)  # 输出 (10,)

# 可选：将数组初始化为0
Z[:] = 0

原因说明 #

• NumPy对(np.float32, 3)这类无名称的子数组dtype，会默认将其解析为“元素是float32，数组额外增加一个维度”，因此自动调整shape和dtype。
• 结构化dtype（即使字段名为空）属于复合类型，NumPy会将其作为整体处理，不会自动展开维度，因此能保留你定义的原始dtype结构。

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