python中数组（numpy.array）的基本操作-Python开发-易学智能

shaoheshaohe 发表于 2020-5-27 08:41:21

python中数组（numpy.array）的基本操作

import numpy as np

数组创建

## 常规创建方法
a = np.array()
b = np.array()
c = np.array([,])
d = np.array([,],dtype=complex) # 指定数据类型
print a, a.dtype
print b, b.dtype
print c, c.dtype
print d, d.dtype

int32
[ 2.3.4.] float64
[[ 1.2.]
[ 3.4.]] float64
[[ 1.+0.j2.+0.j]
[ 3.+0.j4.+0.j]] complex128

shaoheshaohe 发表于 2020-5-27 08:41:37

数组的常用函数
print np.arange(0,7,1,dtype=np.int16) # 0为起点，间隔为1时可缺省(引起歧义下不可缺省)
print np.ones((2,3,4),dtype=np.int16) # 2页，3行，4列，全1，指定数据类型
print np.zeros((2,3,4)) # 2页，3行，4列，全0
print np.empty((2,3)) #值取决于内存
print np.arange(0,10,2) # 起点为0，不超过10，步长为2
print np.linspace(-1,2,5) # 起点为-1，终点为2，取5个点
print np.random.randint(0,3,(2,3)) # 大于等于0，小于3，2行3列的随机整数

[[

]

[

]]
[[[ 0. 0. 0. 0.]
[ 0. 0. 0. 0.]
[ 0. 0. 0. 0.]]

[[ 0. 0. 0. 0.]
[ 0. 0. 0. 0.]
[ 0. 0. 0. 0.]]]
[[ 1.39069238e-309   1.39069238e-309   1.39069238e-309]
[ 1.39069238e-309   1.39069238e-309   1.39069238e-309]]

[-1.   -0.25 0.5   1.25 2. ]
[
]

类型转换
print float(1)
print int(1.0)
print bool(2)
print float(True)



1.0
1
True
1.0

shaoheshaohe 发表于 2020-5-27 08:41:58

数组输出

    从左到右，从上向下
    一维数组打印成行，二维数组打印成矩阵，三维数组打印成矩阵列表

print np.arange(1,6,2)
print np.arange(12).reshape(3,4) # 可以改变输出形状
print np.arange(24).reshape(2,3,4)# 2页，3行，4页



[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[

]]

基本运算

## 元素级运算
a = np.array()
b = np.arange(4)
print a, b
print a-b
print a*b
print a**2
print 2*np.sin(a)
print a>2
print np.exp(a) # 指数



[ 0 2 6 12]
[ 1 4 9 16]
[ 1.68294197 1.81859485 0.28224002 -1.51360499]

[ 2.71828183   7.3890561   20.08553692 54.59815003]

## 矩阵运算（二维数组）
a = np.array([,]) # 2行2列
b = np.arange(6).reshape((2,-1)) # 2行3列
print a,b
print a.dot(b) # 2行3列

[
] [
]
[[ 6 9 12]
]

## 非数组运算，调用方法
a = np.random.randint(0,5,(2,3))
print a
print a.sum(),a.sum(axis=1),a.sum(0) # axis用于指定运算轴（默认全部，可指定0或1）
print a.min(),a.max(axis=1),a.mean(axis=1) # axis = 0: 按列计算，axis = 1: 按行计算
print a.cumsum(1) # 按行计算累积和

[
]
11
0 [ 2.66666667 1.        ]
[
]

shaoheshaohe 发表于 2020-5-27 08:42:17

索引，切片，迭代

## 一维数组
a = np.arange(0,10,1)**2
print a
print a,a,a[-1],a[-2] # 索引从0开始，-1表示最后一个索引
print a,a[-5:-1] # 包括起点，不包括终点
a[-1] = 100; print a # 赋值
a=100; print a # 批量赋值
a[:6:2] = -100; print a # 从开始到第6个索引，每隔一个元素（步长=2）赋值
print a[: :-1];print a # 将a逆序输出，a本身未发生改变
b = ; print b # 通过遍历赋值



[ 0 1 4 9 16 25 36 49 64 81]
0 4 81 64
[ 4 9 16]
[ 0   1   4   9 16 25 36 49 64 100]
[ 0 100 100 100 16 25 36 49 64 100]
[-100 100 -100 100 -100   25   36   49   64 100]
[ 100   64   49   36   25 -100 100 -100 100 -100]
[-100 100 -100 100 -100   25   36   49   64 100]



## 多维数组
a = np.arange(0,20).reshape((4,5))
print a, a, a[:,1], a, a
print a[-1] # 相当于a[-1,:],即索引少于轴数时，确实的索引默认为整个切片

b = np.arange(0,24).reshape((2,3,4))
print b,b # 相当于b 和b
print '-------------------'
for row in a:
    print row # 遍历以第一个轴为基础

[[ 0 1 2 3 4]
[ 5 6 7 8 9]

] 13 [ 1 6 11 16] [ 7 12 17] [[ 5 6 7 8 9]
]

[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[

]]

[

]
-------------------



形状操作

a = np.floor(10*np.random.random((3,4)))
print a, a.shape #输出a的形状
print a.ravel() # 输出平坦化后的a（a本身不改变）
a.shape = (6,2); print a # 改变a的形状
print a.transpose() # 输出a的转置

[[ 0. 4. 3. 2.]
[ 1. 1. 3. 3.]
[ 4. 4. 6. 5.]] (3, 4)
[ 0. 4. 3. 2. 1. 1. 3. 3. 4. 4. 6. 5.]
[[ 0. 4.]
[ 3. 2.]
[ 1. 1.]
[ 3. 3.]
[ 4. 4.]
[ 6. 5.]]
[[ 0. 3. 1. 3. 4. 6.]
[ 4. 2. 1. 3. 4. 5.]]

## 补充：reshape和resize
a = np.array([,])
b = a
a.reshape((3,2))# 不改变数组本身的形状
print a
b.resize((3,2))# 改变数组本身形状
print b



[
]
[

]

shaoheshaohe 发表于 2020-5-27 08:42:38

在numpy模块中，我们经常会使用resize 和 reshape,在具体使用中，通常是使用resize改变数组的尺寸大小，使用reshape用来增加数组的维度。

1.resize

之前看到别人的博客说，resize没有返回值，其实这取决于你如何使用resize,resize有两种使用方式，一种是没有返回值的，直接对原始的数据进行修改，还有一种用法是有返回值的，所以不会修改原有的数组值。

1.1有返回值，不对原始数据进行修改

    import numpy as np
    X=np.array([,
                  ,
                  ])

    X_new=np.resize(X,(3,3)) # do not change the original X
    print("X:\n",X) #original X
    print("X_new:\n",X_new) # new X

    >>
    X:
     [[ 1 2 3 4]
     [ 5 6 7 8]
     [ 9 10 11 12]]
    X_new:
     [

     ]

1.2 无返回值，直接修改原始数组的大小

    import numpy as np
    X=np.array([,
                  ,
                  ])

    X_2=X.resize((3,3)) #change the original X ,and do not return a value
    print("X:\n",X) # change the original X
    print("X_2:\n",X_2) # return None



    X:
     [

     ]
    X_2:
     None

2.reshape

给数组一个新的形状而不改变其数据

    import numpy as np
    X=np.array()

    X_2=X.reshape((2,4)) #retuen a 2*4 2-dim array
    X_3=X.reshape((2,2,2)) # retuen a 2*2*2 3-dim array

    print("X:\n",X)
    print("X_2:\n",X_2)
    print("X_3:\n",X_3)

    >>
    X:

    X_2:
     [
     ]
    X_3:
     [[
      ]

     [
      ]]

shaoheshaohe 发表于 2020-5-27 08:42:43

https://blog.csdn.net/fu6543210/java/article/details/83240024

页: [1]

易学智能's Archiver

python中数组（numpy.array）的基本操作