【numpy笔记_6】数组增删改查和遍历操作

在认识了numpy、创建数组对象和运算的机制后,我们看看如何操作一个数组。


增:

  • 追加末尾:np.append()
  • 在某个索引前插入:np.insert()

与list不同,append和insert方法并非直接在原数组上直接修改,而是返回一个新的数组。

import numpy as np
arr_1 = np.arange(10)
print(f'原始的arr_1: {arr_1}')
res_append = np.append(arr=arr_1, values=[12], axis=0)
res_insert = np.insert(arr=arr_1, obj=0, values=[13], axis=0)
'''append方法【参数】             insert方法【参数】
   arr: 原数组                   arr: 原数组
   (追加末尾)                   obj: 索引位置,在该索引前插入
   values: 插入内容              values: 插入内容
   axis: 插入方向                axis: 插入方向
'''
print(f'操作过append、insert方法后的arr_1: {arr_1}')   # append和insert并未在原数组上修改
print(f'append新赋值的数组:{res_append}')
print(f'insert新赋值的数组:{res_insert}')
# 运行结果:
原始的arr_1: [0 1 2 3 4 5 6 7 8 9]
操作过append、insert方法后的arr_1: [0 1 2 3 4 5 6 7 8 9]
append新赋值的数组:[ 0  1  2  3  4  5  6  7  8  9 12]
insert新赋值的数组:[13  0  1  2  3  4  5  6  7  8  9]
插入方向axis:
arr_1 = np.arange(9).reshape(3,3)
res_append1 = np.append(arr=arr_1, values=[[12,13,14],[15,16,17]],axis=0)  # 插入2行
res_append2 = np.append(arr=arr_1, values=[[12],[13],[14]],axis=1)   # 插入1列
'''
我们知道了shape是(行,列)
axis=0 ———— 沿0轴插入,即插入行(插入完整的一或多行。即插入的数组,其列数要与原数组一致)
axis=1 ———— 沿1轴插入,即插入列(插入完整的一或多列。即插入的数组,其行数要与原数组一致)
'''
print(f'原数组: \n{arr_1}')   # append和insert并未在原数组上修改
print(f'axis=0插入新数组:\n{res_append1}')
print(f'axis=1插入新数组:\n{res_append2}')
# 运行结果:
原数组: 
[[0 1 2]
 [3 4 5]
 [6 7 8]]
axis=0插入新数组:
[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [12 13 14]
 [15 16 17]]
axis=1插入新数组:
[[ 0  1  2 12]
 [ 3  4  5 13]
 [ 6  7  8 14]]

那多维呢?像之前讲的一样,我们理解了shape(x,y,z)和axis参数的关系就像这样:

axis参数的含义.png

插入的值必须按照 axis参数的选择,跟原数组结构保持一致。
来看个多维数组选择axis参数的例子,请仔细观察插入值的数组结构:

arr_1 = np.arange(12).reshape(2,2,3)
res_append1 = np.append(arr=arr_1, values=[[[12,13,14],[15,16,17]]],axis=0)  # 插入1块
res_append2 = np.append(arr=arr_1, values=[[[12,13,14]],[[15,16,17]]],axis=1)   # 插入2行
res_append3 = np.append(arr=arr_1, values=[[[12],[13]],[[14],[15]]],axis=2)   # 插入1列

print(f'原数组: \n{arr_1}')   # append和insert并未在原数组上修改
print(f'axis=0插入新数组:\n{res_append1}')
print(f'axis=1插入新数组:\n{res_append2}')
print(f'axis=2插入新数组:\n{res_append3}')
# 运行结果:
原数组: 
[[[ 0  1  2]
  [ 3  4  5]]

 [[ 6  7  8]
  [ 9 10 11]]]
axis=0插入新数组:
[[[ 0  1  2]
  [ 3  4  5]]

 [[ 6  7  8]
  [ 9 10 11]]

 [[12 13 14]
  [15 16 17]]]
axis=1插入新数组:
[[[ 0  1  2]
  [ 3  4  5]
  [12 13 14]]

 [[ 6  7  8]
  [ 9 10 11]
  [15 16 17]]]
axis=2插入新数组:
[[[ 0  1  2 12]
  [ 3  4  5 13]]

 [[ 6  7  8 14]
  [ 9 10 11 15]]]

再提一点:
axis参数不写默认为None,此时无论数组是什么结构,都会展成一条线将值插入。

arr_1 = np.arange(12).reshape(2,2,3)
res_append1 = np.append(arr=arr_1, values=[12,13,14],axis=None)  # axis参数为None

print(f'原数组: \n{arr_1}')
print(f'axis=0插入新数组:\n{res_append1}')
# 运行结果:
原数组: 
[[[ 0  1  2]
  [ 3  4  5]]

 [[ 6  7  8]
  [ 9 10 11]]]
axis=None插入新数组:
[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14]   # 展成一条线

删:

  • np.delete() 删除指定元素
  • np.unique() 删除重复元素

两个方法仍然并非直接在原数组上直接修改,而是返回一个新的数组。
delete()参数也有arr,obj,axis:

arr_1 = np.arange(40).reshape(2,4,5)
res1 = np.delete(arr=arr_1, obj=5)   # axis参数不选,默认为None,数组会被展开,删除元素5
res2 = np.delete(arr=arr_1, obj=1, axis=2)   # shape(x,y,z)时,axis=2指的是列。即删除第2列
res3 = np.delete(arr=arr_1, obj=[1,3], axis=1)   # shape(x,y,z)时,axis=1指的是行。即删除第2行、第4行
print(f'原数组: \n{arr_1}')
print(f'操作删除 5后的数组:\n{res1}')
print(f'操作删除 第2列后的数组:\n{res2}')
print(f'操作删除 第2行,第4行后的数组:\n{res3}')
# 运行结果:
原数组: 
[[[ 0  1  2  3  4]
  [ 5  6  7  8  9]
  [10 11 12 13 14]
  [15 16 17 18 19]]

 [[20 21 22 23 24]
  [25 26 27 28 29]
  [30 31 32 33 34]
  [35 36 37 38 39]]]
操作删除 5后的数组:
[ 0  1  2  3  4  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39]
操作删除 第2列后的数组:
[[[ 0  2  3  4]
  [ 5  7  8  9]
  [10 12 13 14]
  [15 17 18 19]]

 [[20 22 23 24]
  [25 27 28 29]
  [30 32 33 34]
  [35 37 38 39]]]
操作删除 第2行,第4行后的数组:
[[[ 0  1  2  3  4]
  [10 11 12 13 14]]

 [[20 21 22 23 24]
  [30 31 32 33 34]]]

unique()会将一个数组展开,去重后从小到大排序成一组数。参数主要有:

  • return_index(None、True、False):True则返回新数据元素在原数组中的位置(索引);
  • return_inverse(None、True、False):True则返回原数据元素在新数组中的位置(索引);
  • return_counts(None、True、False):True则返回去重后的数组元素 在原数组中出现的次数。

参数不写,均默认为None。
参数就不再演示了,也好理解。且个人感觉参数用的并不多。

li = [9,4,6,4,5,1,1,9]
arr = np.array(li).reshape(2,4)
arr_1 = np.unique(li)
print(f'原数组: \n{arr}')
print(f'去重后的数组: \n{arr_1}')
# 运行结果:
原数组: 
[[9 4 6 4]
 [5 1 1 9]]
去重后的数组: 
[1 4 5 6 9]

改:

numpy改数据的方法与常规list基本一致,都是通过索引赋值的方式更改。
看几个例子:

arr1 = np.arange(8)
print(f'原数组: \n{arr1}')
arr1[2] = 66   # 列索引,数组为线时代表列
print(f'arr1[2]改后的值: \n{arr1}')

arr1 = np.arange(12).reshape(3,4)
print(f'原数组: \n{arr1}')
arr1[0] = 66   # 行索引,数组为面时代表行
print(f'arr1[0]改后的值: \n{arr1}')
# 上面两个例子看的出,改值索引仍然遵循shape的概念

arr1 = np.arange(12).reshape(3,4)
print(f'原数组: \n{arr1}')
arr1[0:3] = 66   # 行切片索引,左闭右开
print(f'arr1[0:3]改后的值: \n{arr1}')

arr1 = np.arange(12).reshape(3,4)
print(f'原数组: \n{arr1}')
arr1[0,3] = 66   # 坐标索引,数组为面时输入坐标才能定位行、列
print(f'arr1[0,3]改后的值: \n{arr1}')
# 运行结果:
原数组: 
[0 1 2 3 4 5 6 7]
arr1[2]改后的值: 
[ 0  1 66  3  4  5  6  7]
原数组: 
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
arr1[0]改后的值: 
[[66 66 66 66]
 [ 4  5  6  7]
 [ 8  9 10 11]]
原数组: 
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
arr1[0:3]改后的值: 
[[66 66 66 66]
 [66 66 66 66]
 [66 66 66 66]]
原数组: 
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
arr1[0,3]改后的值: 
[[ 0  1  2 66]
 [ 4  5  6  7]
 [ 8  9 10 11]]

查:

上篇《索引、高级索引和切片》中讲过了numpy的索引和切片,根本便是查数据的动作。
这里再介绍几个有意思的方法,在实际场景中使用非常方便。
np.argwhere()方法:加个判断条件,返回数组中某些元素的索引。默认为判断非零元素

li = [0,4,0,2,0,1,9,0]
arr = np.array(li).reshape(2,4)
arr_1 = np.argwhere(arr != 0)   # argwhere()不写也是判断非零的意思
arr_2 = np.argwhere(arr > 4/2)   # argwhere()加其他任何判断条件
print(f'原数组: \n{arr}')
print(f'非零元素的索引: \n{arr_1}')
print(f'大于4/2元素的索引: \n{arr_2}')
# 运行结果:
原数组: 
[[0 4 0 2]
 [0 1 9 0]]
非零元素的索引: 
[[0 1]
 [0 3]
 [1 1]
 [1 2]]
大于4/2元素的索引: 
[[0 1]
 [1 2]]

遍历一个数组:

  • 可以通过for循环遍历;一个for循环只剥离一层,取的是元素块而非最小元素 ;针对多维数组,可以嵌套for循环将元素层层剥离
  • obj.nditer()方法,将数组转成迭代器。取的是数组里的最小元素,每个元素是array对象的形式(该方法直接构建了array对象的迭代器);
  • obj.flat方法(没有括号),将数组转成迭代器。取的是数组里的最小元素,每个元素仅仅只是一个标量(flat把array对象的值取出来,再构建迭代器)。

听着很迷茫,看个例子:

arr1 = np.arange(8).reshape(2,4)

for i in arr1:    # for循环
    print(i)
    print('*'*5)
# 运行结果:
[0 1 2 3]
*****
[4 5 6 7]
*****

arr1 = np.arange(8).reshape(2,4)
arr1 = np.nditer(arr1)   # nditer() 转成迭代器
print(f'nditer迭代器的arr1: {arr1}')
print(f'将arr1转成列表看看里面的元素:{list(arr1)}')
while len(arr1) != 0:
    print(next(arr1))
# 运行结果:
nditer迭代器的arr1: <numpy.nditer object at 0x000002A0D2163710>
将arr1转成列表看看里面的元素:[array(0), array(1), array(2), array(3), array(4), array(5), array(6), array(7)]
0
1
2
3
4
5
6
7

arr1 = np.arange(8).reshape(2,4)
arr1 = arr1.flat   # flat 转成迭代器
print(f'flat迭代器的arr1: {arr1}')
print(f'将arr1转成列表看看里面的元素:{list(arr1)}')
while len(arr1) != 0:
    print(next(arr1))
# 运行结果:
flat迭代器的arr1: <numpy.flatiter object at 0x0000020C1B3D85B0>
将arr1转成列表看看里面的元素:[0, 1, 2, 3, 4, 5, 6, 7]
0
1
2
3
4
5
6
7

看到这你可能明白了刚才说的,obj.nditer()方法,迭代器的每个元素是array对象的形式(该方法直接构建了array对象的迭代器);obj.flat方法,迭代器的每个元素仅仅只是一个标量(flat把array对象的值取出来,再构建迭代器)。

如果flat取的仅仅是值,该迭代器中便是一个个十分纯粹的标量,不能控制读取顺序,更改其元素内容也不能实现原数组的修改。而nditer() 方法恰恰能实现这些:

所以,nditer()可以控制迭代器对象读取元素的顺序,也可以在迭代过程中操作元素的修改。
nditer() 方法参数

  • order='C' (大小写均可)

--- 'C':按行读;
--- 'F':按列读;

  • op_flags=['readwirte']

--- ['readonly']:只读,不能修改
--- ['readwirte']:读写,可读可写

arr = np.arange(8).reshape(2,4)
arr1 = np.nditer(arr, order='c')   # 按行读
try:
    while True:
        print(f'每次提取的元素:{next(arr1)}')   # next()取值遍历
except StopIteration:
    pass
print('* '*20)

arr = np.arange(8).reshape(2,4)
arr2 = np.nditer(arr, order='f', op_flags=['readwrite'])   # 按列读,读写模式,可读可修改
for i in arr2:   # for循环遍历
    if i % 2 == 0:
        i[...] = 9   # [...]是固定写法,指的是改写array对象当前的元素
print(f'修改后的arr: \n{arr}')
# 运行结果:
每次提取的元素:0
每次提取的元素:1
每次提取的元素:2
每次提取的元素:3
每次提取的元素:4
每次提取的元素:5
每次提取的元素:6
每次提取的元素:7
* * * * * * * * * * * * * * * * * * * * 
修改后的arr: 
[[9 1 9 3]
 [9 5 9 7]]

最后总结一下遍历:

推荐使用nditer()方法,能控制遍历的顺序,以及遍历时对元素进行操作。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 219,589评论 6 508
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,615评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 165,933评论 0 356
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,976评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,999评论 6 393
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,775评论 1 307
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,474评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,359评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,854评论 1 317
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,007评论 3 338
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,146评论 1 351
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,826评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,484评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,029评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,153评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,420评论 3 373
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,107评论 2 356

推荐阅读更多精彩内容