python之numpy的基本使用

学无涯愈进而愈惘

2018-08-17

关注关注

一、numpy概述

numpy（Numerical Python）提供了python对多维数组对象的支持：ndarray，具有矢量运算能力，快速、节省空间。numpy支持高级大量的维度数组与矩阵运算，此外也针对数组运算提供大量的数学函数库。

二、创建ndarray数组

ndarray：N维数组对象（矩阵），所有元素必须是相同类型。

ndarray属性：ndim属性，表示维度个数；shape属性，表示各维度大小；dtype属性，表示数据类型。

创建ndarray数组函数：

python之numpy的基本使用

代码示例：

# -*- coding: utf-8 -*-

import numpy;

print '使用列表生成一维数组'

data = [1,2,3,4,5,6]

x = numpy.array(data)

print x #打印数组

print x.dtype #打印数组元素的类型

print '使用列表生成二维数组'

data = [[1,2],[3,4],[5,6]]

x = numpy.array(data)

print x #打印数组

print x.ndim #打印数组的维度

print x.shape #打印数组各个维度的长度。shape是一个元组

print '使用zero/ones/empty创建数组:根据shape来创建'

x = numpy.zeros(6) #创建一维长度为6的，元素都是0一维数组

print x

x = numpy.zeros((2,3)) #创建一维长度为2，二维长度为3的二维0数组

print x

x = numpy.ones((2,3)) #创建一维长度为2，二维长度为3的二维1数组

print x

x = numpy.empty((3,3)) #创建一维长度为2，二维长度为3,未初始化的二维数组

print x

print '使用arrange生成连续元素'

print numpy.arange(6) # [0,1,2,3,4,5] 开区间

print numpy.arange(0,6,2) # [0, 2，4]

三、指定ndarray数组元素的类型

NumPy数据类型：

python之numpy的基本使用

代码示例：

print '生成指定元素类型的数组:设置dtype属性'

x = numpy.array([1,2.6,3],dtype = numpy.int64)

print x # 元素类型为int64

print x.dtype

x = numpy.array([1,2,3],dtype = numpy.float64)

print x # 元素类型为float64

print x.dtype

print '使用astype复制数组，并转换类型'

x = numpy.array([1,2.6,3],dtype = numpy.float64)

y = x.astype(numpy.int32)

print y # [1 2 3]

print x # [ 1. 2.6 3. ]

z = y.astype(numpy.float64)

print z # [ 1. 2. 3.]

print '将字符串元素转换为数值元素'

x = numpy.array(['1','2','3'],dtype = numpy.string_)

y = x.astype(numpy.int32)

print x # ['1' '2' '3']

print y # [1 2 3] 若转换失败会抛出异常

print '使用其他数组的数据类型作为参数'

x = numpy.array([ 1., 2.6,3. ],dtype = numpy.float32);

y = numpy.arange(3,dtype=numpy.int32);

print y # [0 1 2]

print y.astype(x.dtype) # [ 0. 1. 2.]

四、ndarray的矢量化计算

矢量运算：相同大小的数组键间的运算应用在元素上

矢量和标量运算：“广播”— 将标量“广播”到各个元素

代码示例：

print 'ndarray数组与标量/数组的运算'

x = numpy.array([1,2,3])

print x*2 # [2 4 6]

print x>2 # [False False True]

y = numpy.array([3,4,5])

print x+y # [4 6 8]

print x>y # [False False False]

五、ndarray数组的基本索引和切片

一维数组的索引：与Python的列表索引功能相似

多维数组的索引：

arr[r1:r2, c1:c2]
arr[1,1] 等价 arr[1][1]
[:] 代表某个维度的数据

代码示例：

print 'ndarray的基本索引'

x = numpy.array([[1,2],[3,4],[5,6]])

print x[0] # [1,2]

print x[0][1] # 2,普通python数组的索引

print x[0,1] # 同x[0][1]，ndarray数组的索引

x = numpy.array([[[1, 2], [3,4]], [[5, 6], [7,8]]])

print x[0] # [[1 2],[3 4]]

y = x[0].copy() # 生成一个副本

z = x[0] # 未生成一个副本

print y # [[1 2],[3 4]]

print y[0,0] # 1

y[0,0] = 0

z[0,0] = -1

print y # [[0 2],[3 4]]

print x[0] # [[-1 2],[3 4]]

print z # [[-1 2],[3 4]]

print 'ndarray的切片'

x = numpy.array([1,2,3,4,5])

print x[1:3] # [2,3] 右边开区间

print x[:3] # [1,2,3] 左边默认为 0

print x[1:] # [2,3,4,5] 右边默认为元素个数

print x[0:4:2] # [1,3] 下标递增2

x = numpy.array([[1,2],[3,4],[5,6]])

print x[:2] # [[1 2],[3 4]]

print x[:2,:1] # [[1],[3]]

x[:2,:1] = 0 # 用标量赋值

print x # [[0,2],[0,4],[5,6]]

x[:2,:1] = [[8],[6]] # 用数组赋值

print x # [[8,2],[6,4],[5,6]]

六、ndarray数组的布尔索引和花式索引

布尔索引：使用布尔数组作为索引。arr[condition]，condition为一个条件/多个条件组成的布尔数组。

布尔型索引代码示例：

print 'ndarray的布尔型索引'

x = numpy.array([3,2,3,1,3,0])

# 布尔型数组的长度必须跟被索引的轴长度一致

y = numpy.array([True,False,True,False,True,False])

print x[y] # [3,3,3]

print x[y==False] # [2,1,0]

print x>=3 # [ True False True False True False]

print x[~(x>=3)] # [2,1,0]

print (x==2)|(x==1) # [False True False True False False]

print x[(x==2)|(x==1)] # [2 1]

x[(x==2)|(x==1)] = 0

print x # [3 0 3 0 3 0]

花式索引：使用整型数组作为索引。

花式索引代码示例：

print 'ndarray的花式索引:使用整型数组作为索引'

x = numpy.array([1,2,3,4,5,6])

print x[[0,1,2]] # [1 2 3]

print x[[-1,-2,-3]] # [6,5,4]

x = numpy.array([[1,2],[3,4],[5,6]])

print x[[0,1]] # [[1,2],[3,4]]

print x[[0,1],[0,1]] # [1,4] 打印x[0][0]和x[1][1]

print x[[0,1]][:,[0,1]] # 打印01行的01列 [[1,2],[3,4]]

# 使用numpy.ix_()函数增强可读性

print x[numpy.ix_([0,1],[0,1])] #同上打印01行的01列 [[1,2],[3,4]]

x[[0,1],[0,1]] = [0,0]

print x # [[0,2],[3,0],[5,6]]

七、ndarray数组的转置和轴对换

数组的转置/轴对换只会返回源数据的一个视图，不会对源数据进行修改。

代码示例：

print 'ndarray数组的转置和轴对换'

k = numpy.arange(9) #[0,1,....8]

m = k.reshape((3,3)) # 改变数组的shape复制生成2维的，每个维度长度为3的数组

print k # [0 1 2 3 4 5 6 7 8]

print m # [[0 1 2] [3 4 5] [6 7 8]]

# 转置(矩阵)数组：T属性 : mT[x][y] = m[y][x]

print m.T # [[0 3 6] [1 4 7] [2 5 8]]

# 计算矩阵的内积 xTx

print numpy.dot(m,m.T) # numpy.dot点乘

# 高维数组的轴对象

k = numpy.arange(8).reshape(2,2,2)

print k # [[[0 1],[2 3]],[[4 5],[6 7]]]

print k[1][0][0]

# 轴变换 transpose 参数:由轴编号组成的元组

m = k.transpose((1,0,2)) # m[y][x][z] = k[x][y][z]

print m # [[[0 1],[4 5]],[[2 3],[6 7]]]

print m[0][1][0]

# 轴交换 swapaxes (axes：轴)，参数:一对轴编号

m = k.swapaxes(0,1) # 将第一个轴和第二个轴交换 m[y][x][z] = k[x][y][z]

print m # [[[0 1],[4 5]],[[2 3],[6 7]]]

print m[0][1][0]

# 使用轴交换进行数组矩阵转置

m = numpy.arange(9).reshape((3,3))

print m # [[0 1 2] [3 4 5] [6 7 8]]

print m.swapaxes(1,0) # [[0 3 6] [1 4 7] [2 5 8]]

八、ndarray通用函数

通用函数（ufunc）是一种对ndarray中的数据执行元素级运算的函数。

一元ufunc：

python之numpy的基本使用

一元ufunc代码示例：

print '一元ufunc示例'

x = numpy.arange(6)

print x # [0 1 2 3 4 5]

print numpy.square(x) # [ 0 1 4 9 16 25]

x = numpy.array([1.5,1.6,1.7,1.8])

y,z = numpy.modf(x)

print y # [ 0.5 0.6 0.7 0.8]

print z # [ 1. 1. 1. 1.]

二元ufunc：

python之numpy的基本使用

二元ufunc代码示例：

print '二元ufunc示例'

x = numpy.array([[1,4],[6,7]])

y = numpy.array([[2,3],[5,8]])

print numpy.maximum(x,y) # [[2,4],[6,8]]

print numpy.minimum(x,y) # [[1,3],[5,7]]

九、NumPy的where函数使用

np.where(condition, x, y)，第一个参数为一个布尔数组，第二个参数和第三个参数可以是标量也可以是数组。

代码示例：

print 'where函数的使用'

cond = numpy.array([True,False,True,False])

x = numpy.where(cond,-2,2)

print x # [-2 2 -2 2]

cond = numpy.array([1,2,3,4])

x = numpy.where(cond>2,-2,2)

print x # [ 2 2 -2 -2]

y1 = numpy.array([-1,-2,-3,-4])

y2 = numpy.array([1,2,3,4])

x = numpy.where(cond>2,y1,y2) # 长度须匹配

print x # [1,2,-3,-4]

print 'where函数的嵌套使用'

y1 = numpy.array([-1,-2,-3,-4,-5,-6])

y2 = numpy.array([1,2,3,4,5,6])

y3 = numpy.zeros(6)

cond = numpy.array([1,2,3,4,5,6])

x = numpy.where(cond>5,y3,numpy.where(cond>2,y1,y2))

print x # [ 1. 2. -3. -4. -5. 0.]

十、ndarray常用的统计方法

可以通过这些基本统计方法对整个数组/某个轴的数据进行统计计算。

python之numpy的基本使用

代码示例：

print 'numpy的基本统计方法'

x = numpy.array([[1,2],[3,3],[1,2]]) #同一维度上的数组长度须一致

print x.mean() # 2

print x.mean(axis=1) # 对每一行的元素求平均

print x.mean(axis=0) # 对每一列的元素求平均

print x.sum() #同理 12

print x.sum(axis=1) # [3 6 3]

print x.max() # 3

print x.max(axis=1) # [2 3 2]

print x.cumsum() # [ 1 3 6 9 10 12]

print x.cumprod() # [ 1 2 6 18 18 36]

用于布尔数组的统计方法：

sum : 统计数组/数组某一维度中的True的个数
any：统计数组/数组某一维度中是否存在一个/多个True
all：统计数组/数组某一维度中是否都是True

代码示例：

print '用于布尔数组的统计方法'

x = numpy.array([[True,False],[True,False]])

print x.sum() # 2

print x.sum(axis=1) # [1,1]

print x.any(axis=0) # [True,False]

print x.all(axis=1) # [False,False]

使用sort对数组/数组某一维度进行就地排序（会修改数组本身）。

代码示例：

print '.sort的就地排序'

x = numpy.array([[1,6,2],[6,1,3],[1,5,2]])

x.sort(axis=1)

print x # [[1 2 6] [1 3 6] [1 2 5]]

#非就地排序：numpy.sort()可产生数组的副本

十一、ndarray数组的去重以及集合运算

python之numpy的基本使用

代码示例：（方法返回类型为一维数组（1d）)

print 'ndarray的唯一化和集合运算'

x = numpy.array([[1,6,2],[6,1,3],[1,5,2]])

print numpy.unique(x) # [1,2,3,5,6]

y = numpy.array([1,6,5])

print numpy.in1d(x,y) # [ True True False True True False True True False]

print numpy.setdiff1d(x,y) # [2 3]

print numpy.intersect1d(x,y) # [1 5 6]

十二、numpy中的线性代数

import numpy.linalg 模块。线性代数（linear algebra）

常用的numpy.linalg模块函数：

python之numpy的基本使用

代码示例：

print '线性代数'

import numpy.linalg as nla

print '矩阵点乘'

x = numpy.array([[1,2],[3,4]])

y = numpy.array([[1,3],[2,4]])

print x.dot(y) # [[ 5 11][11 25]]

print numpy.dot(x,y) # # [[ 5 11][11 25]]

print '矩阵求逆'

x = numpy.array([[1,1],[1,2]])

y = nla.inv(x) # 矩阵求逆（若矩阵的逆存在）

print x.dot(y) # 单位矩阵 [[ 1. 0.][ 0. 1.]]

print nla.det(x) # 求行列式

十三、numpy中的随机数生成

import numpy.random模块。

常用的numpy.random模块函数：

python之numpy的基本使用

代码示例：

print 'numpy.random随机数生成'

import numpy.random as npr

x = npr.randint(0,2,size=100000) #抛硬币

print (x>0).sum() # 正面的结果

print npr.normal(size=(2,2)) #正态分布随机数数组 shape = (2,2)

十四、ndarray数组重塑

代码示例:

print 'ndarray数组重塑'

x = numpy.arange(0,6) #[0 1 2 3 4]

print x #[0 1 2 3 4]

print x.reshape((2,3)) # [[0 1 2][3 4 5]]

print x #[0 1 2 3 4]

print x.reshape((2,3)).reshape((3,2)) # [[0 1][2 3][4 5]]

y = numpy.array([[1,1,1],[1,1,1]])

x = x.reshape(y.shape)

print x # [[0 1 2][3 4 5]]

print x.flatten() # [0 1 2 3 4 5]

x.flatten()[0] = -1 # flatten返回的是拷贝

print x # [[0 1 2][3 4 5]]

print x.ravel() # [0 1 2 3 4 5]

x.ravel()[0] = -1 # ravel返回的是视图（引用）

print x # [[-1 1 2][3 4 5]]

print "维度大小自动推导"

arr = numpy.arange(15)

print arr.reshape((5, -1)) # 15 / 5 = 3

十五、ndarray数组的拆分与合并

python之numpy的基本使用

代码示例：

print '数组的合并与拆分'

x = numpy.array([[1, 2, 3], [4, 5, 6]])

y = numpy.array([[7, 8, 9], [10, 11, 12]])

print numpy.concatenate([x, y], axis = 0)

# 竖直组合 [[ 1 2 3][ 4 5 6][ 7 8 9][10 11 12]]

print numpy.concatenate([x, y], axis = 1)

# 水平组合 [[ 1 2 3 7 8 9][ 4 5 6 10 11 12]]

print '垂直stack与水平stack'

print numpy.vstack((x, y)) # 垂直堆叠:相对于垂直组合

print numpy.hstack((x, y)) # 水平堆叠：相对于水平组合

# dstack：按深度堆叠

print numpy.split(x,2,axis=0)

# 按行分割 [array([[1, 2, 3]]), array([[4, 5, 6]])]

print numpy.split(x,3,axis=1)

# 按列分割 [array([[1],[4]]), array([[2],[5]]), array([[3],[6]])]

# 堆叠辅助类

import numpy as np

arr = np.arange(6)

arr1 = arr.reshape((3, 2))

arr2 = np.random.randn(3, 2)

print 'r_用于按行堆叠'

print np.r_[arr1, arr2]

'''

[[ 0. 1. ]

[ 2. 3. ]

[ 4. 5. ]

[ 0.22621904 0.39719794]

[-1.2201912 -0.23623549]

[-0.83229114 -0.72678578]]

'''

print 'c_用于按列堆叠'

print np.c_[np.r_[arr1, arr2], arr]

'''

[[ 0. 1. 0. ]

[ 2. 3. 1. ]

[ 4. 5. 2. ]

[ 0.22621904 0.39719794 3. ]

[-1.2201912 -0.23623549 4. ]

[-0.83229114 -0.72678578 5. ]]

'''

print '切片直接转为数组'

print np.c_[1:6, -10:-5]

'''

[[ 1 -10]

[ 2 -9]

[ 3 -8]

[ 4 -7]

[ 5 -6]]

'''

十六、数组的元素重复操作

代码示例：

print '数组的元素重复操作'

x = numpy.array([[1,2],[3,4]])

print x.repeat(2) # 按元素重复 [1 1 2 2 3 3 4 4]

print x.repeat(2,axis=0) # 按行重复 [[1 2][1 2][3 4][3 4]]

print x.repeat(2,axis=1) # 按列重复 [[1 1 2 2][3 3 4 4]]

x = numpy.array([1,2])

print numpy.tile(x,2) # tile瓦片：[1 2 1 2]

print numpy.tile(x, (2, 2)) # 指定从低维到高维依次复制的次数。

# [[1 2 1 2][1 2 1 2]]

全部代码：Github

python数组 python 数组

学无涯愈进而愈惘

0 关注 0 粉丝 0 动态

关注关注

python如何获得list或numpy数组中最大元素对应的索引

#表示最大值在第二行第二列

sizhixht 2020-11-16

Python数据分析Numpy库常用函数详解，提到循环就该想到的库

Python进行数据分析的核心库肯定是Pandas，该库差不多可以解决结构化数据的绝大部分处理需求。在《Python数据分析常用函数及参数详解，可以留着以备不时之需》一文中也已经对该库的常用函数进行了详细介绍。但是Pandas是构建在Numpy的基础之

CodeAsWind 2020-08-17

【Python数据科学】Numpy速查

Numpy是python数据科学计算的核心库，提供高性能的多维数组对象及处理数组的工具。np.zeros #创建值为0的数组。np.linspace #创建均匀间

sunnyhappy0 2020-07-09

python 把一个数组arr按照指定的数组大小size分割成若干个数组块

python 把一个数组arr按照指定的数组大小size分割成若干个数组块这里的运行结果，我是以分割2个数组块的。def arr_size: s=[] for i in range: c=arr[i:i+size]

cxcxrs 2020-06-26

python--之数组(array)

petal0 2020-06-10

数组中重复的数字

　　在一个长度为n的数组里的所有数字都在0到n-1的范围内。例如，如果输入长度为7的数组{2,3,1,0,2,5,3}，那么对应第一个重复的数字是2。　　第一种思路：寻找重复的数字最容易想到的就是先排序后查找，因为如果数字出现重复，那么排序后相邻的数字必定

深思千年 2020-06-10

利用Python进行数据分析

Numpy ndarray: 多维数组对象ndarray属性ndarray包含的每一个元素均为相同类型。shape属性：数组每一维度的数量dtype属性：数组的数据类型ndim属性：数组的维数NumPy 数据类型numpy 支持的数据类型比 Python

xmwang0 2020-06-01

JS基础

// alert(‘哈哈哈‘);alert弹框// 在htmle我们通过script标签src方法进行导入console.log(‘学习啊‘);//单行注释/*多行注释 */// js语言是以分号为结束符// 声明一个变量,需要var 关键字+变量名=‘‘

fgleeldq 2020-05-27

YAML

YAML 是 "YAML Ain‘t a Markup Language"的递归缩写。在开发的这种语言时，YAML 的意思其实是："Yet Another Markup Language"。YAML 的语法和其他高级

ShiShuo 2020-05-16

用实践带领你进入numpy的世界——(二)：numpy基本数组创建函数

np.zeros()创建全零数组np.empty()创建数值不定的数组np.ones()创建全一数组np.linspace()创建任意长度的一维数组np.arange()与python中的range使用方法相同，返回的是一个数组np.array()创建标准

文山羊 2020-05-08

pandas_一维数组与常用操作

# 一维数组与常用操作。# 设置输出结果列对齐。# 使用字典创建 Series 字典的键作为索引。# 修改 Series 对象的值。s2[‘语文‘] = 94. # 将 s1 所有的值都加 5、使用加法时，对所有元素都进行。# 在 s1 的索引下标前加入参

tengyuan 2020-05-07

LeetCode小白入门——简单题目八题合集，每题两解

本文共包括八个题目，来源于LeetCode简单难度，每个问题会给出两种解法，第一种偏暴力、易理解一些，第二种会更加高效一些，尽可能会避免利用Python的内置函数，便于真正理解算法原理。和一个目标值 target，请你在该数组中找出和为目标值的那两整数，并

Leewoxinyiran 2020-05-06

LeetCode 45. 跳跃游戏 II | Python

给定一个非负整数数组，你最初位于数组的第一个位置。数组中的每个元素代表你在该位置可以跳跃的最大长度。你的目标是使用最少的跳跃次数到达数组的最后一个位置。从下标为 0 跳到下标为 1 的位置，跳 1 步，然后跳 3 步到达数组的最后一个位置。这题要求与之前

sschencn 2020-05-04

numpy

# np.arange()可以类比 range(), 前者生成一个ndarray对象,后者生成range()对象。# 使用array创建三维,四维.....# np.random.randn 返回一个或一组样本,具有标准正态分布。a = np.random

vevoly 2020-05-04

go语言3 数组, 切片, Maps

//1 基本使用：定义。//定义了一个大小为3的int类型数组。//var a [3]int //只定义，没有初始化。//只给第2个位置设为99. // go语言中，都是copy传递。python中都是引用传递，一切皆对象，就是地址，当做参数传递是把

一只码畜 2020-04-22

python之numpy多维数组

NumPy 是 Python 语言的一个扩展程序库，支持大量的维度数组与矩阵运算，此外也针对数组运算提供大量的数学函数库。NumPy它本身其实没有提供很高级别的数据分析功能，NumPy之于数值计算特别重要的原因之一，调用astype总会创建一个新的数组，即

sulindong0 2020-03-09

Python之Numpy

...a.shape=2,2,3 #改成2组2行3列三维结构[[[0,1,2] [3,4,5]][[6,7,8] [9,10,11]]]

一只码畜 2020-03-01

Go语言中的数组与数组切片

　　　　　　　　　　　　　　　　　　　　fmt.Printlb. 所有的值类型变量在赋值和作为参数传递的时候都将进行一次复制，会将副本传递过去。所以函数体中无法通过修改传入的数组内容来改变数组。　　　　　　｛1.一个指向原生数组的指针；　　　　　　　3.数

GoatSucker 2020-02-24

【Python数组及其基础操作】【Numpy ndarray】

　　在python中创建数组最简单的办法就是使用array函数。它接受一切序列型的对象，然后产生一个含有传入数据的numpy数组。其中，嵌套序列会被转换为一个多维数组。np.random.normal(1.7, 0.1, (2, 3)0 #给定均值/标准

sdwylry 2020-02-14

numpy学习（一）

　　小编整理了一些numpy常用的函数，也是方便小编以后查看。　　常用的有dtype，ndim，写参数时要指名。np.bool #Boolean type storing TRUE and FALSE values. 　　np.dype也可以实现上述的内容

mieleizhi0 2020-02-01

安科网

python之numpy的基本使用

学无涯愈进而愈惘

学无涯愈进而愈惘

相关推荐

python如何获得list或numpy数组中最大元素对应的索引

Python数据分析Numpy库常用函数详解，提到循环就该想到的库

【Python数据科学】Numpy速查

python 把一个数组arr按照指定的数组大小size分割成若干个数组块

python--之数组(array)

数组中重复的数字

利用Python进行数据分析

JS基础

YAML

用实践带领你进入numpy的世界——(二)：numpy基本数组创建函数

pandas_一维数组与常用操作

LeetCode小白入门——简单题目八题合集，每题两解

LeetCode 45. 跳跃游戏 II | Python

numpy

go语言3 数组, 切片, Maps

python之numpy多维数组

Python之Numpy

Go语言中的数组与数组切片

【Python数组及其基础操作】【Numpy ndarray】

numpy学习（一）

学无涯愈进而愈惘