作者：刘早起

来源：早起Python

大家好，又到了numpy进阶修炼专题。numpy大家应该不陌生了，看了太多的原理讲解之后，用刷题来学习是最有效的方法，本文将带来20个NumPy经典问题，附赠20段实用代码，拿走就用，建议打开Jupyter Notebook边敲边看！

01数据查找

问：如何获得两个数组之间的相同元素

输入：

import numpy as np
import pandas as pd
import warnings
warnings.filterwarnings("ignore")
arr1 = np.random.randint(10,6,6)
arr2 = np.random.randint(10,6,6)

答案：

arr1 = np.random.randint(10,6,6)
arr2 = np.random.randint(10,6,6)
print("arr1: %s"%arr1)
print("arr2: %s"%arr2)
np.intersect1d(arr1,arr2)

02数据修改

问:如何从一个数组中删除另一个数组存在的元素

输入：

arr1 = np.random.randint(10,6,6)
arr2 = np.random.randint(10,6,6)

答案：

arr1 = np.random.randint(1,10,10)
arr2 = np.random.randint(1,10,10)
print("arr1: %s"%arr1)
print("arr2: %s"%arr2)
np.setdiff1d(arr1,arr2)

03数据修改

问：如何修改一个数组为只读模式

输入：

arr1 = np.random.randint(1,10,10)

答案：

arr1 = np.random.randint(1,10,10)
arr1.flags.writeable = False

04数据转换

问：如何将list转为numpy数组

输入：

a = [1,2,3,4,5]

答案：

a = [1,2,3,4,5]
np.array(a)

05数据转换

问：如何将pd.DataFrame转为numpy数组

输入：

df = pd.DataFrame({'A':[1,2,3],'B':[4,5,6],'C':[7,8,9]})

答案：

df.values

06数据分析

问：如何使用numpy进行描述性统计分析

输入：

arr1 = np.random.randint(1,10,10)
arr2 = np.random.randint(1,10,10)

答案：

arr1 = np.random.randint(1,10,10)
arr2 = np.random.randint(1,10,10)

print("arr1的平均数为:%s" %np.mean(arr1))
print("arr1的中位数为:%s" %np.median(arr1))
print("arr1的方差为:%s" %np.var(arr1))
print("arr1的标准差为:%s" %np.std(arr1))
print("arr1,arr的相关性矩阵为:%s" %np.cov(arr1,arr2))
print("arr1,arr的协方差矩阵为:%s" %np.corrcoef(arr1,arr2))

07数据抽样

问：如何使用numpy进行概率抽样

arr = np.array([1,2,3,4,5])

输入：

arr = np.array([1,2,3,4,5])
np.random.choice(arr,10,p = [0.1,0.1,0.1,0.1,0.6])

答案：

08数据创建

问：如何为数据创建副本

输入：

arr = np.array([1,2,3,4,5])

答案：

#对副本数据进行修改，不会影响到原始数据
arr = np.array([1,2,3,4,5])
arr1 = arr.copy()

09数据切片

问：如何对数组进行切片

输入：

arr = np.arange(10)

备注：从索引2开始到索引8停止，间隔为2

答案：

arr = np.arange(10)
a = slice(2,8,2)
arr[a] #等价于arr[2:8:2]

10字符串操作

问：如何使用NumPy操作字符串

输入：

str1 = ['I love']
str2 = [' Python']

答案：

#拼接字符串
str1 = ['I love']
str2 = [' Python']
print(np.char.add(str1,str2))

#大写首字母
str3 = np.char.add(str1,str2)
print(np.char.title(str3))

以上就是我总结的NumPy经典20题中的10题，你都会吗？并且每题我都只给出了一种解法，而事实上每题都有多种解法，所以你应该思考是否有更好的思路，下一篇继续给你列出另外10题哈！

CDA学员免费下载查看报告全文：2026全球数智化人才指数报告【CDA数据科学研究院】.pdf