Python数据分析学习笔记5——NumPy数组操作

off999 2024-11-06 11:22 38 浏览 0 评论

形状修改

语法：arr.reshape(m,n)

将数组修改成m行n列的新数组

# -*- coding:utf-8 -*-
import numpy as np

print('一维数组')
nums = [1,2,3,4,5,6,7,8]   #一维数组
arr = np.array(nums)
arr1 = arr.reshape(2,4)
print('原数组\n',arr)
print('变形后数组\n',arr1)

print('一维数组')
nums = [[1,2,3,4],[5,6,7,8]]  #二维数组
arr = np.array(nums)
arr1 = arr.reshape(4,2)
print('原数组\n',arr)
print('变形后数组\n',arr1)

维度修改

语法：numpy.array(arr, ndmin=n)

ndmin=n 表示将数组修改为n维数组
ndmin参数只能将低维度数组转换成高维数组，反之不行

# -*- coding:utf-8 -*-
import numpy as np

print('低维转高维')
arr = np.array([1,2,3])
result1 = np.array(arr,ndmin =2)
result2 = np.array(arr,ndmin =3)
print('原数组\n',arr)
print('转换为二维数组\n',result1)
print('转换为三维数组\n',result2)

print('高维转低维')

arr = np.array([[1,2,3],[4,5,6]])
result1 = np.array(arr,ndmin =1)
print('原数组\n',arr)
print('转换后的数组\n',result1)
print('如上结果高维转低维并未成功')

数组翻转

语法：numpy.transpose(arr)

将数组翻转，m*n →n*m

# -*- coding:utf-8 -*-
import numpy as np

print('数组翻转')
arr = np.arange(1,21).reshape(4,5)
print('原数组\n',arr)
arr1 = np.transpose(arr)
print('翻转后的数组\n',arr1)

数组去重

语法： numpy.unique(arr)

unique()返回的是一个一维数组。

# -*- coding:utf-8 -*-
import numpy as np

print('一维数组')
arr = np.array([1,1,2,2,3,3,4,4,5,5,6,6])
print('原数组\n',arr)
arr1 = np.unique(arr)
print('去重后的数组\n',arr1)

print('二维数组')
arr = np.array([[1,1,2,2,3,3,7],[4,4,5,5,6,6,7]])
print('原数组\n',arr)
arr1 = np.unique(arr)
print('去重后的数组\n',arr1)

print('列表去重')
nums =[1,2,3,4,5,4,3,2,1]
arr = np.array(nums)
result = list(np.unique(arr))
print(result)

数组合并

语法：numpy.concatenate((arr1,arr2),axis=n)

沿“现有轴”合并两个数组
arr1和arr2是形状相同，元素类型相同的数组

# -*- coding:utf-8 -*-
import numpy as np

arr1 = np.array([[1,2,3],[4,5,6]])
arr2 = np.array([[7,8,9],[10,11,12]])

print('原数组arr1\n',arr1)
print('原数组arr2\n',arr2)

print('沿着 纵轴 合并,垂直堆叠 等价于 vstack()')
result1 = np.concatenate((arr1,arr2),axis = 0)
print('合并的数组\n',result1)
print('沿着 横轴 合并,水平堆叠  等价于 hstack()')
result2 = np.concatenate((arr1,arr2),axis = 1)
print('合并的数组\n',result2)

语法：

numpy.hstack((arr1,arr2)) 沿“现有轴”水平合并

numpy.vstack((arr1,arr2)) 沿“现有轴”垂直合并

# -*- coding:utf-8 -*-
import numpy as np

arr1 = np.array([[1,2,3],[4,5,6]])
arr2 = np.array([[7,8,9],[10,11,12]])

print('原数组arr1\n',arr1)
print('原数组arr2\n',arr2)

print('沿着 纵轴 合并')
result1 = np.hstack((arr1,arr2))
print('合并的数组\n',result1)
print('沿着 横轴 合并')
result2 = np.vstack((arr1,arr2))
print('合并的数组\n',result2)

语法：numpy.stack((arr1,arr2),axis=n)

沿“新的轴”合并（堆叠）两个数组
arr1和arr2是形状相同，元素类型相同的数组
axis用于指定沿哪一条轴操作
合并后的数组维度会更大

# -*- coding:utf-8 -*-
import numpy as np

arr1 = np.array([[1,2,3],[4,5,6]])
arr2 = np.array([[7,8,9],[10,11,12]])

print('原数组arr1\n',arr1)
print('原数组arr2\n',arr2)

print('沿着 纵轴 合并')
result1 = np.stack((arr1,arr2),axis = 0)
print('合并的数组,合并后是个三维数组\n',result1)
print('沿着 横轴 合并')
result2 = np.stack((arr1,arr2),axis = 1)
print('合并的数组,合并后是个三维数组\n',result2)

数组分割

语法：numpy.split(arr,section = 整数或数组， axis = n)

section 为整数：表示用该整数进行平均分割
section为数组：表示沿着轴的位置进行分割（左开有闭）
axis = 0 表示沿着纵轴分割(垂直分割) 等价于 np.vsplit(arr,section =m）
axis = 1 表示沿着纵轴分割(垂直分割) 等价于 np.hsplit(arr,section =m）

# -*- coding:utf-8 -*-
import numpy as np
print('一维数组')
arr = np.arange(10)
print('原数组arr\n',arr)
result1 = np.split(arr,5)
print('分割后的数组(section=5)\n',result1)
print('类型',type(result1))
result2 = np.split(arr,2)
print('分割后的数组(section=2)\n',result2)
print('类型',type(result2))

# -*- coding:utf-8 -*-
import numpy as np


print('二维数组')
arr = np.arange(10).reshape(5,2)
print('原数组arr\n',arr)
result1 = np.split(arr,5,axis=0)
result2 = np.vsplit(arr,5)
print('axis = 0 表示沿着纵轴分割(垂直分割) 等价于 np.vsplit(arr,2)')
print('分割后的数组(section=5,axis=0)\n',result1)
print('分割后的数组(vsplit)\n',result2)

print('---------分割线----------')
result3 = np.split(arr,2,axis=1)
result4 = np.hsplit(arr,2)
print('axis = 1 表示沿着横轴分割(水平分割) 等价于 np.hsplit(arr,2)')
print('分割后的数组(section=2,axis=1)\n',result3)
print('分割后的数组(hsplit)\n',result4)

# -*- coding:utf-8 -*-
import numpy as np

print('section 是数组')
arr = np.arange(12)
print('原数组arr\n',arr)
result1 = np.split(arr,[1,2,3],axis=0)
print('分割后的数组(是数组,axis=0)\n',result1)
result2 = np.split(arr,[1,2,4],axis=0)
print('分割后的数组(是数组,axis=0)\n',result2)

学习参考资料：《从0到1Python数据分析》

python数组去重