python dtype

import numpy as np

# 创建一个包含整数的NumPy数组，并指定数据类型为32位整数
arr = np.array([1, 2, 3, 4], dtype=np.int32)
print("数组:", arr)
print("数据类型:", arr.dtype)

# 创建一个包含浮点数的NumPy数组，并指定数据类型为64位浮点数
float_arr = np.array([1.5, 2.5, 3.5, 4.5], dtype=np.float64)
print("浮点数数组:", float_arr)
print("浮点数数据类型:", float_arr.dtype)

# 尝试将字符串转换为整数，如果失败会抛出错误
try:
    str_arr = np.array(['1', '2', 'three'], dtype=np.int32)
except ValueError as e:
    print("转换错误:", e)

# 强制转换数据类型，即使会导致数据丢失
str_to_int_arr = np.array(['1', '2', '3'], dtype=np.int32)
print("字符串转整数数组:", str_to_int_arr)

解释说明：

• dtype 参数用于指定 NumPy 数组中元素的数据类型。
• np.int32 表示 32 位整数类型，np.float64 表示 64 位浮点数类型。
• 如果尝试将不兼容的数据（如字符串 "three"）转换为整数，会抛出 ValueError。
• 使用 dtype 参数可以确保数组中的所有元素都具有相同的数据类型，这有助于优化内存使用和计算效率。