1.数组_代码示例
import numpy as np
import random
#使用numpy生成数组,得到ndarray的类型
t1 = np.array([1,3,2,5])
print(type(t1))
print(t1)
#使用array中的range()
t2 = np.array(range(2, 10))
print(type(t2)) # <class 'numpy.ndarray'>
print(t2) # [1 3 2 5]
t3 = np.arange(1, 10, 2)
print(type(t3)) # <class 'numpy.ndarray'>
print(t3) # [2 3 4 5 6 7 8 9]
print(t3.dtype) # t3数据类型
print('*'*10) # int32
#numpy中的数据类型
t4 = np.array(range(1, 4), dtype="i1") # "i1"表示int8
print(t4) # [1 2 3]
print(t4.dtype) # int8
##numpy中的bool类型
t5 = np.array([1,1,0,1,0,0],dtype=bool)
print(t5) # [ True True False True False False]
print(t5.dtype) # bool
#调整数据类型
t6 = t5.astype("int8")
print(t6) # [1 1 0 1 0 0]
print(t6.dtype) # int8
#numpy中的小数
t7 = np.array([random.random() for i in range(3)])
print(t7) # [0.96033818 0.63804794 0.82357959]
print(t7.dtype) # float64
t8 = np.round(t7, 2)
print(t8) # [0.96 0.64 0.82]
2.数组形状_代码示例
import numpy as np
a = np.array([[1,2,3], [4,5,6]])
print(a.shape) # (2, 1)
print(a.reshape(3, 2)) # [[1, 2], [3, 4], [5, 6]]
print(a.reshape(1, 6)) # [[1, 2, 3, 4, 5, 6]]
print(a.flatten()) # [1, 2, 3, 4, 5, 6] 一维数组
print(a+1) # [[2, 3, 4], [5, 6, 7]] 广播机制
print(a*2) # [[ 2, 4, 6], [ 8, 10, 12]]
b = np.array([1, 2, 3])
print(a+b) # [[2, 4, 6], [5, 7, 9]]
print(a*b) # [[ 1, 4, 9], [ 4, 10, 18]]
print(a-b) # [[0, 0, 0], [3, 3, 3]]
3.csv读取_代码示例
import numpy as np
# ./表示当前路径
us_file_path = "./youtube_video_data/US_video_data_numbers.csv"
uk_file_path = "./youtube_video_data/GB_video_data_numbers.csv"
# delimiter分割字符串, dtype数据类型, unpack是否转置
t1 = np.loadtxt(us_file_path, delimiter=",", dtype="int", unpack