Numpy学习笔记（介绍）

最新推荐文章于 2024-05-18 18:22:14 发布

闪电磊

最新推荐文章于 2024-05-18 18:22:14 发布

阅读量677

点赞数

分类专栏： Python 文章标签： python 数据分析机器学习

本文链接：https://blog.csdn.net/chenleismr/article/details/104261019

版权

Python 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

什么是 NumPy?
Numpy是一个Python库/模块，在Python编程中用于科学计算。在本教程中，你将学习如何在Numpy数组上执行一系列操作，例如以多种方式添加、删除、排序和操作元素。
官方网站：https://www.numpy.org.cn/
NumPy是使用Python进行科学计算的基础软件包。除其他外，它包括：
功能强大的N维数组对象。
精密广播功能函数。
集成 C/C+和Fortran 代码的工具。
强大的线性代数、傅立叶变换和随机数功能。
Numpy提供一个多维数组对象和其他派生数组，如掩码数组或掩码多维数组。

Numpy是Python中科学计算的基础包。它是一个Python库，提供多维数组对象，各种派生对象（如掩码数组和矩阵），以及用于数组快速操作的各种API，有包括数学、逻辑、形状操作、排序、选择、输入输出、离散傅立叶变换、基本线性代数，基本统计运算和随机模拟等等。

为什么使用Numpy？
Numpy模块提供一个ndarray对象，我们可以用这个对象来对任意维度的数组执行操作。ndarray代表N维数组，其中N是任意数字。这意味着Numpy数组可以是任意维度的数组。
NumPy 最重要的一个特点是其 N 维数组对象 ndarray，它是一系列同类型数据的集合，以 0 下标为开始进行集合中元素的索引。ndarray 对象是用于存放同类型元素的多维数组。ndarray 中的每个元素在内存中都有相同存储大小的区域。
与Python列表相比，Numpy有很多优点。我们可以对Numpy数组执行高性能操作，例如：

数组成员排序
数学和逻辑运算
输入/输出功能
统计和线性代数运算

添加数组元素
你可以使用NumPy模块的append（）方法添加一个NumPy数组元素。append的使用操作如下：
numpy.append(array, value, axis)
这些值将附加在数组的末尾，新的ndarray将与上面所示的新值和旧值一起返回。

axis是一个可选的整数，用于定义数组的显示方式。如果没有指定axis，数组结构将被展平，如你稍后将看到的一样。

请看下面的示例，首先创建一个数组，然后使用append方法向该数组添加更多值：
a = numpy.array([1, 2, 3])

newArray = numpy.append (a, [10, 11, 12])

添加列
使用Numpy的append（）方法插入列。
请看下面的示例，创建了一个二维数组并插入了两列：

a = numpy.array([[1, 2, 3], [4, 5, 6]])
b = numpy.array([[400], [800]])
newArray = numpy.append(a, b, axis = 1)
如果未使用axis属性，则输出如下：
这就是数组结构的扁平化过程。
在NumPy中，还可以使用insert（）方法插入元素或列。insert（）和append（）两种方法之间的区别在于我们在使用insert（）方法时，可以指定要在哪个索引处添加元素，而append（）方法则会在数组末尾添加一个值。
请看以下示例：

a = numpy.array([1, 2, 3])
newArray = numpy.insert(a, 1, 90)
这里，insert（）方法在索引1处添加元素。请记住数组索引从0开始。

添加行
在本节中，我们将使用append（）方法向数组添加行。此方法就像在数组中添加元素一样简单。请看以下示例：

a = numpy.array([[1, 2, 3], [4, 5, 6]])
newArray = numpy.append(a, [[50, 60, 70]], axis = 0)

利器之一：切片和索引
ndarray对象的内容可以通过索引或切片来访问和修改，与 Python 中 list 的切片操作一样。ndarray 数组可以基于 0 - n 的下标进行索引，切片对象可以通过内置的 slice 函数，并设置 start, stop 及 step 参数进行，从原数组中切割出一个新数组。

一个栗子

import numpy as np
a = np.arange(15).reshape(3, 5)
a
array([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14]])

a.shape
(3, 5)

a.ndim # 数组轴的个数，在python的世界中，轴的个数被称作秩
2

a.dtype.name
‘int64’

a.itemsize # 数组中每个元素的字节大小。
8

a.size
15

type(a)
<type ‘numpy.ndarray’>

b = np.array([6, 7, 8])
b
array([6, 7, 8])

type(b)
<type ‘numpy.ndarray’>
创建矩阵
对于Python中的numpy模块，一般用其提供的ndarray对象。创建一个ndarray对象很简单，只要将一个list作为参数即可。例如:

import numpy as np
#创建一维的narray对象

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

a.dtype
dtype(‘int64’)

浮点类型

b = np.array([1.2, 3.5, 5.1])
b.dtype
dtype(‘float64’)
#创建二维的narray对象

a2 = np.array([[1,2,3,4,5],[6,7,8,9,10]])
b = np.array([(1.5,2,3), (4,5,6)]) # 使用的是元组
b
array([[ 1.5, 2. , 3. ],
[ 4. , 5. , 6. ]])

The type of the array can also be explicitly specified at creation time:

c = np.array( [ [1,2], [3,4] ], dtype=complex )
c
array([[ 1.+0.j, 2.+0.j],
[ 3.+0.j, 4.+0.j]])
矩阵行数列数
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a.shape) #结果返回一个tuple元组 (2, 5) 2行，5列
print(a.shape[0]) #获得行数，返回 2
print(a.shape[1]) #获得列数，返回 5
矩阵按行列选取
矩阵的截取和list相同，可以通过[]（方括号）来截取

import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a[0:1]) #截取第一行,返回 [[1 2 3 4 5]]
print(a[1,2:5]) #截取第二行，第三、四、五列，返回 [8 9 10]
print(a[1,:]) #截取第二行,返回 [ 6 7 8 9 10]
矩阵按条件截取
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
b = a[a>6] # 截取矩阵a中大于6的元素，返回的是一维数组
print(b) # 返回 [ 7 8 9 10]

其实布尔语句首先生成一个布尔矩阵，将布尔矩阵传入[]（方括号）实现截取

print(a>6)

[[False False False False False]
[False True True True True]]
按条件截取应用较多的是对矩阵中满足一定条件的元素变成特定的值。例如将矩阵中大于6的元素变成0。

import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a)
#开始矩阵为
[[ 1 2 3 4 5]
[ 6 7 8 9 10]]

a[a>6] = 0
print(a)
#大于6清零后矩阵为
[[1 2 3 4 5]
[6 0 0 0 0]]
Stacking together different arrays
矩阵的合并可以通过numpy中的hstack方法和vstack方法实现:

a = np.floor(10*np.random.random((2,2)))
a
array([[ 8., 8.],
[ 0., 0.]])

b = np.floor(10*np.random.random((2,2)))
b
array([[ 1., 8.],
[ 0., 4.]])

np.vstack((a,b))
array([[ 8., 8.],
[ 0., 0.],
[ 1., 8.],
[ 0., 4.]])

np.hstack((a,b))
array([[ 8., 8., 1., 8.],
[ 0., 0., 0., 4.]])
矩阵的合并也可以通过concatenatef方法。
np.concatenate( (a1,a2), axis=0 ) 等价于 np.vstack( (a1,a2) )
np.concatenate( (a1,a2), axis=1 ) 等价于 np.hstack( (a1,a2) )

通过函数创建矩阵
arange
import numpy as np
a = np.arange(10) # 默认从0开始到10（不包括10），步长为1
print(a) # 返回 [0 1 2 3 4 5 6 7 8 9]
a1 = np.arange(5,10) # 从5开始到10（不包括10），步长为1
print(a1) # 返回 [5 6 7 8 9]
a2 = np.arange(5,20,2) # 从5开始到20（不包括20），步长为2
print(a2) # 返回 [ 5 7 9 11 13 15 17 19]
linspace/ logspace
import numpy as np

类似于matlab

a = np.linspace(0,10,7) # 生成首位是0，末位是10，含7个数的等差数列
print(a)

结果

[ 0. 1.66666667 3.33333333 5. 6.66666667 8.33333333 10. ]

a = np.ones(3, dtype=np.int32)
b = np.linspace(0,pi,3)
b.dtype.name
‘float64’

c = a+b
c
array([ 1. , 2.57079633, 4.14159265])

c.dtype.name
‘float64’

d = np.exp(c*1j)
d
array([ 0.54030231+0.84147098j, -0.84147098+0.54030231j,
-0.54030231-0.84147098j])

d.dtype.name
‘complex128’
ones、zeros、eye、empty
ones创建全1矩阵，zeros创建全0矩阵，eye创建单位矩阵，empty创建空矩阵（实际有值）

import numpy as np

a_ones = np.ones((3,4)) # 创建3*4的全1矩阵
print(a_ones)

结果

[[ 1. 1. 1. 1.]
[ 1. 1. 1. 1.]
[ 1. 1. 1. 1.]]

np.ones((2,3,4), dtype=np.int16 ) # dtype can also be specified
array([[[ 1, 1, 1, 1],
[ 1, 1, 1, 1],
[ 1, 1, 1, 1]],
[[ 1, 1, 1, 1],
[ 1, 1, 1, 1],
[ 1, 1, 1, 1]]], dtype=int16)

a_zeros = np.zeros((3,4)) # 创建3*4的全0矩阵
print(a_zeros)

结果

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

a_eye = np.eye(3) # 创建3阶单位矩阵
print(a_eye)

结果

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

a_empty = np.empty((3,4)) # 创建3*4的空矩阵
print(a_empty)

结果

[[ 1.78006111e-306 -3.13259416e-294 4.71524461e-309 1.94927842e+289]
[ 2.10230387e-309 5.42870216e+294 6.73606381e-310 3.82265219e-297]
[ 6.24242356e-309 1.07034394e-296 2.12687797e+183 6.88703165e-315]]

有些矩阵太大，如果不想省略中间部分，通过set_printoptions来强制NumPy打印所有数据。

np.set_printoptions(threshold=‘nan’)
fromstring
fromstring()方法可以将字符串转化成ndarray对象，需要将字符串数字化时这个方法比较有用，可以获得字符串的ascii码序列。

import numpy as np

a = “abcdef”
b = np.fromstring(a,dtype=np.int8) # 因为一个字符为8位，所以指定dtype为np.int8
print(b) # 返回 [ 97 98 99 100 101 102]
random

a = np.random.random((2,3)) # 产生2行,3列的随机矩阵
a
array([[ 0.18626021, 0.34556073, 0.39676747],
[ 0.53881673, 0.41919451, 0.6852195 ]])
fromfunction
fromfunction()方法可以根据矩阵的行号列号生成矩阵的元素。例如创建一个矩阵，矩阵中的每个元素都为行号和列号的和。

import numpy as np

def func(i,j):
return i+j

a = np.fromfunction(func,(5,6))

第一个参数为指定函数，第二个参数为列表list或元组tuple,说明矩阵的大小

print(a)

[[ 0. 1. 2. 3. 4. 5.]
[ 1. 2. 3. 4. 5. 6.]
[ 2. 3. 4. 5. 6. 7.]
[ 3. 4. 5. 6. 7. 8.]
[ 4. 5. 6. 7. 8. 9.]]
#注意这里行号的列号都是从0开始的
矩阵的运算
常用矩阵运算符
Numpy中的ndarray对象重载了许多运算符，使用这些运算符可以完成矩阵间对应元素的运算。

运算符说明+矩阵对应元素相加-矩阵对应元素相减*矩阵对应元素相乘/矩阵对应元素相除，如果都是整数则取商%矩阵对应元素相除后取余数矩阵每个元素都取n次方，如2：每个元素都取平方

import numpy as np
a1 = np.array([[4,5,6],[1,2,3]])
a2 = np.array([[6,5,4],[3,2,1]])

print(a1+a2) # 相加

结果

[[10 10 10]
[ 4 4 4]]

print(a1/a2) # 整数相除取商

结果

[[0 1 1]
[0 1 3]]

print(a1%a2) # 相除取余数

结果

[[4 0 2]
[1 0 0]]
常用矩阵函数
同样地，numpy中也定义了许多函数，使用这些函数可以将函数作用于矩阵中的每个元素。表格中默认导入了numpy模块，即 import numpy as np 。a为ndarray对象。

常用矩阵函数说明np.sin(a)对矩阵a中每个元素取正弦,sin(x)np.cos(a)对矩阵a中每个元素取余弦,cos(x)np.tan(a)对矩阵a中每个元素取正切,tan(x)np.arcsin(a)对矩阵a中每个元素取反正弦,arcsin(x)np.arccos(a)对矩阵a中每个元素取反余弦,arccos(x)np.arctan(a)对矩阵a中每个元素取反正切,arctan(x)np.exp(a)对矩阵a中每个元素取指数函数,exnp.sqrt(a)对矩阵a中每个元素开根号

当矩阵中的元素不在函数定义域范围内，会产生RuntimeWarning，结果为nan(not a number)
矩阵乘法（点乘）
矩阵乘法必须满足矩阵乘法的条件，即第一个矩阵的列数等于第二个矩阵的行数。矩阵乘法的函数为 dot 。

import numpy as np

a1 = np.array([[1,2,3],[4,5,6]]) # a1为23矩阵
a2 = np.array([[1,2],[3,4],[5,6]]) # a2为32矩阵

print(a1.shape[1]==a2.shape[0]) # True, 满足矩阵乘法条件
print(a1.dot(a2))

a1.dot(a2)相当于matlab中的a1*a2

而Python中的a1a2相当于matlab中的a1.a2

结果

[[22 28]
[49 64]]
矩阵的转置 a.T
import numpy as np
a = np.array([[1,2,3],[4,5,6]])
print(a.transpose())

结果

[[1 4]
[2 5]
[3 6]]
矩阵的转置还有更简单的方法，就是a.T。

import numpy as np

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

结果

[[1 4]
[2 5]
[3 6]]
矩阵的逆
设A是数域上的一个n阶方阵，若在相同数域上存在另一个n阶矩阵B，使得： AB=BA=E。则我们称B是A的逆矩阵，而A则被称为可逆矩阵。
求矩阵的逆需要先导入numpy.linalg，用linalg的inv函数来求逆。矩阵求逆的条件是矩阵应该是方阵。

import numpy as np
import numpy.linalg as lg

a = np.array([[1,2,3],[4,5,6],[7,8,9]])

print(lg.inv(a))

结果

[[ -4.50359963e+15 9.00719925e+15 -4.50359963e+15]
[ 9.00719925e+15 -1.80143985e+16 9.00719925e+15]
[ -4.50359963e+15 9.00719925e+15 -4.50359963e+15]]

a = np.eye(3) # 3阶单位矩阵
print(lg.inv(a)) # 单位矩阵的逆为他本身

结果

[[ 1. 0. 0.]
[ 0. 1. 0.]
[ 0. 0. 1.]]
矩阵信息获取（如均值等）
最值
获得矩阵中元素最大最小值的函数分别是max和min，可以获得整个矩阵、行或列的最大最小值。

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
print(a.max()) #获取整个矩阵的最大值结果： 6
print(a.min()) #结果：1

可以指定关键字参数axis来获得行最大（小）值或列最大（小）值

axis=0 行方向最大（小）值，即获得每列的最大（小）值

axis=1 列方向最大（小）值，即获得每行的最大（小）值

例如

print(a.max(axis=0))

结果为 [4 5 6]

print(a.max(axis=1))

结果为 [3 6]

要想获得最大最小值元素所在的位置，可以通过argmax函数来获得

print(a.argmax(axis=1))

结果为 [2 2]

平均值
获得矩阵中元素的平均值可以通过函数mean()。同样地，可以获得整个矩阵、行或列的平均值。

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
print(a.mean()) #结果为： 3.5

同样地，可以通过关键字axis参数指定沿哪个方向获取平均值

print(a.mean(axis=0)) # 结果 [ 2.5 3.5 4.5]
print(a.mean(axis=1)) # 结果 [ 2. 5.]
方差
方差的函数为var(),方差函数var()相当于函数mean(abs(x - x.mean())**2),其中x为矩阵。

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
print(a.var()) # 结果 2.91666666667

print(a.var(axis=0)) # 结果 [ 2.25 2.25 2.25]
print(a.var(axis=1)) # 结果 [ 0.66666667 0.66666667]
标准差
标准差的函数为std()。 std()相当于sqrt(mean(abs(x - x.mean())**2))，或相当于sqrt(x.var())。

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
print(a.std()) # 结果 1.70782512766

print(a.std(axis=0)) # 结果 [ 1.5 1.5 1.5]
print(a.std(axis=1)) # 结果 [ 0.81649658 0.81649658]
中值
中值指的是将序列按大小顺序排列后，排在中间的那个值，如果有偶数个数，则是排在中间两个数的平均值。中值的函数是median()，调用方法为numpy.median(x,[axis])，axis可指定轴方向，默认axis=None，对所有数取中值。

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

print(np.median(x)) # 对所有数取中值

结果

3.5

print(np.median(x,axis=0)) # 沿第一维方向取中值

结果

[ 2.5 3.5 4.5]

print(np.median(x,axis=1)) # 沿第二维方向取中值

结果

[ 2. 5.]
求和
矩阵求和的函数是sum()，可以对行，列，或整个矩阵求和

import numpy as np

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

print(a.sum()) # 对整个矩阵求和

结果 21

print(a.sum(axis=0)) # 对行方向求和

结果 [5 7 9]

print(a.sum(axis=1)) # 对列方向求和

结果 [ 6 15]

累积和
某位置累积和指的是该位置之前(包括该位置)所有元素的和。例如序列[1,2,3,4,5]，其累计和为[1,3,6,10,15]，即第一个元素为1，第二个元素为1+2=3，……，第五个元素为1+2+3+4+5=15。矩阵求累积和的函数是cumsum()，可以对行，列，或整个矩阵求累积和。

import numpy as np

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

print(a.cumsum()) # 对整个矩阵求累积和

结果 [ 1 3 6 10 15 21]

print(a.cumsum(axis=0)) # 对列方向求累积和

结果

[[1 2 3]
[5 7 9]]

print( a.cumsum(axis=1)) # 对行方向求累积和

结果

[[ 1 3 6]
[ 4 9 15]]
极差

import numpy as np
a = np.arange(10)
a.ptp()

结果是

9
百分位数
numpy.percentile(a, q, axis)
序号参数及描述1.a 输入数组2.q 要计算的百分位数，在 0 ~ 100 之间3.axis 沿着它计算百分位数的轴

加权平均值

data = range(1,5)
data
[1, 2, 3, 4]

np.average(data)
2.5

np.average(range(1,11), weights=range(10,0,-1))
4.0

data = np.arange(6).reshape((3,2))
data
array([[0, 1],
[2, 3],
[4, 5]])

np.average(data, axis=1, weights=[1./4, 3./4])
array([ 0.75, 2.75, 4.75])

np.average(data, weights=[1./4, 3./4])
Traceback (most recent call last):
…
TypeError: Axis must be specified when shapes of a and weights differ.
Shape Manipulation
Changing the shape of an array

a = np.floor(10*np.random.random((3,4)))
a
array([[ 2., 8., 0., 6.],
[ 4., 5., 1., 1.],
[ 8., 9., 3., 6.]])

a.shape
(3, 4)
数组的形状可以用以下方式改变。Note that the following three commands all return a modified array, but do not change the original array:

a.ravel() # returns the array, flattened
array([ 2., 8., 0., 6., 4., 5., 1., 1., 8., 9., 3., 6.])

a.reshape(6,2) # returns the array with a modified shape
array([[ 2., 8.],
[ 0., 6.],
[ 4., 5.],
[ 1., 1.],
[ 8., 9.],
[ 3., 6.]])

a.T # returns the array, transposed
array([[ 2., 4., 8.],
[ 8., 5., 9.],
[ 0., 1., 3.],
[ 6., 1., 6.]])

a.T.shape
(4, 3)

a.shape
(3, 4)
The reshape function returns its argument with a modified shape, whereas thendarray.resize method modifies the array itself:

a
array([[ 2., 8., 0., 6.],
[ 4., 5., 1., 1.],
[ 8., 9., 3., 6.]])

a.resize((2,6))
a
array([[ 2., 8., 0., 6., 4., 5.],
[ 1., 1., 8., 9., 3., 6.]])
If a dimension is given as -1 in a reshaping operation, the other dimensions are automatically calculated:

a.reshape(3,-1)
array([[ 2., 8., 0., 6.],
[ 4., 5., 1., 1.],
[ 8., 9., 3., 6.]])
Splitting one array into several smaller ones
Using hsplit, you can split an array along its horizontal axis, either by specifying the number of equally shaped arrays to return, or by specifying the columns after which the division should occur:

a = np.floor(10*np.random.random((2,12)))
a
array([[ 9., 5., 6., 3., 6., 8., 0., 7., 9., 7., 2., 7.],
[ 1., 4., 9., 2., 2., 1., 0., 6., 2., 2., 4., 0.]])

np.hsplit(a,3) # Split a into 3
[array([[ 9., 5., 6., 3.],
[ 1., 4., 9., 2.]]), array([[ 6., 8., 0., 7.],
[ 2., 1., 0., 6.]]), array([[ 9., 7., 2., 7.],
[ 2., 2., 4., 0.]])]
Copies and Views
When operating and manipulating arrays, their data is sometimes copied into a new array and sometimes not. This is often a source of confusion for beginners. There are three cases:

No Copy At All
a = b,改变b就相当于改变a,或者相反。

a = np.arange(12)
b = a # no new object is created
b is a # a and b are two names for the same ndarray object
True

b.shape = 3,4 # changes the shape of a
a.shape
(3, 4)
View or Shallow Copy
Different array objects can share the same data. The view method creates a new array object that looks at the same data.

c = a.view()
c is a
False

c.base is a # c is a view of the data owned by a
True

c.flags.owndata
False

c.shape = 2,6 # a’s shape doesn’t change
a.shape
(3, 4)

c[0,4] = 1234 # a’s data changes
a
array([[ 0, 1, 2, 3],
[1234, 5, 6, 7],
[ 8, 9, 10, 11]])
Slicing an array returns a view of it:

s = a[:,1:3] # spaces added for clarity; could also be written “s = a[:,1:3]”
s[:] = 10 # s[:] is a view of s. Note the difference between s=10 and s[:]=10
a
array([[ 0, 10, 10, 3],
[1234, 10, 10, 7],
[ 8, 10, 10, 11]])
Deep Copy
The copy method makes a complete copy of the array and its data.

d = a.copy() # a new array object with new data is created
d is a
False

d.base is a # d doesn’t share anything with a
False

d[0,0] = 9999
a
array([[ 0, 10, 10, 3],
[1234, 10, 10, 7],
[ 8, 10, 10, 11]])
numpy关于copy有三种情况，完全不复制、视图（view）或者叫浅复制（shadow copy）和深复制（deep copy）。而b = a[:]就属于第二种，即视图，这本质上是一种切片操作（slicing），所有的切片操作返回的都是视图。具体来说，b = a[:]会创建一个新的对象b（所以说 id 和a不一样），但是b的数据完全来自于a，和a保持完全一致，换句话说，b的数据完全由a保管，他们两个的数据变化是一致的，可以看下面的示例:

a = np.arange(4) # array([0, 1, 2, 3])
b = a[:] # array([0, 1, 2, 3])

b.flags.owndata # 返回 False，b 并不保管数据
a.flags.owndata # 返回 True，数据由 a 保管

改变 a 同时也影响到 b

a[-1] = 10 # array([0, 1, 2, 10])
b # array([0, 1, 2, 10])

改变 b 同时也影响到 a

b[0] = 10 # array([10, 1, 2, 10])
a # array([10, 1, 2, 10])
b = a 和 b = a[:] 的差别就在于后者会创建新的对象，前者不会。两种方式都会导致a和b的数据相互影响。要想不让a的改动影响到b，可以使用深复制：unique_b = a.copy()

曼德勃罗
import numpy as np
import matplotlib.pyplot as plt
def mandelbrot( h,w, maxit=20 ):
“”“Returns an image of the Mandelbrot fractal of size (h,w).”""
y,x = np.ogrid[ -1.4:1.4:h1j, -2:0.8:w1j ]
c = x+y*1j
z = c
divtime = maxit + np.zeros(z.shape, dtype=int)

for i in range(maxit):
    z = z**2 + c
    diverge = z*np.conj(z) > 2**2            # who is diverging
    div_now = diverge & (divtime==maxit)  # who is diverging now
    divtime[div_now] = i                  # note when
    z[diverge] = 2                        # avoid diverging too much

return divtime

plt.imshow(mandelbrot(400,400))
plt.show()