numpy 01

61 阅读1分钟
import time

import numpy as np
from twisted.python.util import println

print("Hello World!")

np_arr = np.arange(1000000)
start = time.time()
np.sum(np_arr**2)
print(f"NumPy数组耗时: {time.time()-start:.4f}s")

arr = np.array([[1, 2], [3, 4], [5, 6]])
print("形状 shape:", arr.shape)
print("维度 ndim:", arr.ndim)
print("元素个数 size:", arr.size)
print("元素类型 dtype:", arr.dtype)

# NumPy数组耗时: 0.0029s
# 形状 shape: (3, 2)
# 维度 ndim: 2
# 元素个数 size: 6
# 元素类型 dtype: int64

println("转置:行变列,列变行 T",arr.T)
println("单个元素占用的内存字节数 itemsize",arr.itemsize)
println("数组占用的内存字节数 nbytes",arr.nbytes)

# 转置:行变列,列变行 T [[1 3 5]
# [2 4 6]]
# 单个元素占用的内存字节数 itemsize 8
# 数组占用的内存字节数 nbytes 48
print("内存存储方式 bytes",arr.flags)