Numpy 常用数据结构和清理函数

381 阅读2分钟

「这是我参与11月更文挑战的第16天,活动详情查看:2021最后一次更文挑战

Numpy 清理工具

Numpy常用数据结构


  • Numpy中常用的数据结构是ndarray格式
  • 使用array函数创建,语法格式为array(列表或元组)
  • 可以使用其他函数例如arange、linspace、zeros等创建
import numpy as np
arr1 = np.array([-9, 7, 4, 3])
arr1
array([-9,  7,  4,  3])
type(arr1)  # n维数组
numpy.ndarray
arr1 = np.array([-9, 7, 4, 3], dtype='str')
arr1
array(['-9', '7', '4', '3'], dtype='<U2')
arr1 = np.array([-9, 7, 4, 3], dtype=float)
arr1
array([-9.,  7.,  4.,  3.])
arr1 = np.array([-9, 7, 4, 3], dtype=int)
arr1
array([-9,  7,  4,  3])
arr2 = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
arr2
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])
for i in range(1, 10):
    print(i)
1
2
3
4
5
6
7
8
9
np.arange(1, 10, 0.5)
array([1. , 1.5, 2. , 2.5, 3. , 3.5, 4. , 4.5, 5. , 5.5, 6. , 6.5, 7. ,
       7.5, 8. , 8.5, 9. , 9.5])
# 等差数组
# 第一个参数:起始值
# 第二个参数:终止值
# 第三个参数:元素个数
# endpoint:是否包含终值
np.linspace(1, 10, 10, endpoint=True)
array([ 1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10.])
np.linspace(1, 10, 20, endpoint=True)
array([ 1.        ,  1.47368421,  1.94736842,  2.42105263,  2.89473684,
        3.36842105,  3.84210526,  4.31578947,  4.78947368,  5.26315789,
        5.73684211,  6.21052632,  6.68421053,  7.15789474,  7.63157895,
        8.10526316,  8.57894737,  9.05263158,  9.52631579, 10.        ])
9/19  # 步长
0.47368421052631576
1 + 3 * (9/19)
2.4210526315789473
# 产生一个4行5列的数组,值为0
np.zeros([4, 5])
array([[0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0.]])
np.zeros(4)
array([0., 0., 0., 0.])
# 产生一个2行3列的数组,值为1
np.ones([2, 3])
array([[1., 1., 1.],
       [1., 1., 1.]])
# 对每一个数组元素加1
arr2 + 1
array([[ 2,  3,  4,  5],
       [ 6,  7,  8,  9],
       [10, 11, 12, 13]])
# 判断数组的维数
arr1.ndim
1
arr2.ndim
2
# 判断数组的形状
arr1.shape
(4,)
arr2.shape
(3, 4)
# 返回数组元素个数
arr2.size
12
# 返回数组元素类型
arr2.dtype
dtype('int32')
data2 = ((8.5, 6, 4, 1.2, 0.7), (1.5, 3, 5.4, 7.3, 9), (3.2, 4.5, 6, 3, 9), (11.2, 13.4, 15.6, 17.8, 19))
arr3 = np.array(data2)
arr3
array([[ 8.5,  6. ,  4. ,  1.2,  0.7],
       [ 1.5,  3. ,  5.4,  7.3,  9. ],
       [ 3.2,  4.5,  6. ,  3. ,  9. ],
       [11.2, 13.4, 15.6, 17.8, 19. ]])
arr3[0]
array([8.5, 6. , 4. , 1.2, 0.7])
arr3[3]
array([11.2, 13.4, 15.6, 17.8, 19. ])
# 取第二行第三列元素
arr3[1, 2]
5.4
arr3[1][2]
5.4
arr3[:, 3]
array([ 1.2,  7.3,  3. , 17.8])
# 取第二列到第三列元素
arr3[:, 1:3]
array([[ 6. ,  4. ],
       [ 3. ,  5.4],
       [ 4.5,  6. ],
       [13.4, 15.6]])
arr3[3][1]
13.4

数组下标从0开始,且左闭右开\color{red}数组下标从0开始,且左闭右开

Numpy常用数据清理函数

import numpy as np
s = np.array([1,2,3,4,3,2,1,2,2,4,6,7,2,4,8,4,5])
s = np.sort(s)
s
array([1, 1, 2, 2, 2, 2, 2, 3, 3, 4, 4, 4, 4, 5, 6, 7, 8])
np.array(sorted(s, reverse=True))
array([8, 7, 6, 5, 4, 4, 4, 4, 3, 3, 2, 2, 2, 2, 2, 1, 1])
# 返回排序的索引
np.argsort(s)
array([ 0,  6, 12,  7,  5,  8,  1,  2,  4,  3, 15,  9, 13, 16, 10, 11, 14],
      dtype=int64)
arr1 = np.array([[0,1,3],[4,2,9],[4,5,9],[1,-3,4]])
# axis=0:表示对列排序
# axis=1:表示对行排序
np.sort(arr1, axis=0)
array([[ 0, -3,  3],
       [ 1,  1,  4],
       [ 4,  2,  9],
       [ 4,  5,  9]])
np.sort(arr1, axis=1)
array([[ 0,  1,  3],
       [ 2,  4,  9],
       [ 4,  5,  9],
       [-3,  1,  4]])
s
array([1, 2, 3, 4, 3, 2, 1, 2, 2, 4, 6, 7, 2, 4, 8, 4, 5])
# 第一个参数:条件
# 第二个参数:条件满足的返回值
# 第三个参数:条件不满足的返回值
# 大于3返回元素本身,不大于3返回-1
np.where(s>3, s, -1)
array([-1, -1, -1,  4, -1, -1, -1, -1, -1,  4,  6,  7, -1,  4,  8,  4,  5])
# 第一个参数:条件
# 第二个参数:返回的值
# 筛选数组中值大于3的元素
np.extract(s > 3, s)
array([4, 4, 6, 7, 4, 8, 4, 5])

最后,欢迎大家关注我的个人微信公众号 『小小猿若尘』,获取更多IT技术、干货知识、热点资讯