python内置函数之enumerate和sum函数

1 enumerate简介

格式
enumerate(iterable, start=0)
功能
返回一个枚举对象, 其中iterable必须是序列,迭代器或其他支持迭代的对象.迭代器的__next__()方法的返回值,通过enumerate返回,值类型为tuple,包括从start(默认为0)技术的数字和迭代器包含的数据.

2 案例

2.1 遍历一维列表

li = ["tianlanlan", "yunbaibai", "shuiqingqing"]
en_li = enumerate(li)
print("enumerate list: {}".format(list(en_li)))
for idx, value in enumerate(li):
    print("idx: {} value: {}".format(idx, value))
1
2
3
4
5

• Result

enumerate list: [(0, 'tianlanlan'), (1, 'yunbaibai'), (2, 'shuiqingqing')]
idx: 0 value: tianlanlan
idx: 1 value: yunbaibai
idx: 2 value: shuiqingqing
1
2
3
4

• Analysis
  (1) 通过enumerate返回一个tuple,序号从0开始,个数为迭代器数据个数,值为迭代器数据;
  (2) 实现数据的key:value对应.

2.2 遍历二维列表

x_sample = np.random.rand(4, 3)
print("x_sample: {}".format(x_sample))
for idx, value in enumerate(x_sample):
    print("idx: {} value: {}".format(idx, value))
1
2
3
4

• Result

x_sample: [[0.73356379 0.46344017 0.12340443]
 [0.96689566 0.70792431 0.87034476]
 [0.73511355 0.77274115 0.05571193]
 [0.56135686 0.33162721 0.82986378]]
idx: 0 value: [0.73356379 0.46344017 0.12340443]
idx: 1 value: [0.96689566 0.70792431 0.87034476]
idx: 2 value: [0.73511355 0.77274115 0.05571193]
idx: 3 value: [0.56135686 0.33162721 0.82986378]
1
2
3
4
5
6
7
8

• Analysis
  (1) 二维数组[4, 3]即4行三列,enumerate将的编号为0,1,2,3个数为行数,数据为列中存储的数据;
  (2) 对于数组,最终需要获取数组中的数据,行列的作用是索引功能,enumerate按照行建立数组索引获取数据;
  (2) 对于二维数组,key即编号使用行数,value使用对应的列;

2.3 遍历三维数组

import numpy as np
z_sample = np.random.uniform(-1, 1, size=(2 , 3, 3))
print("z sample: {}".format(z_sample))
for idx, value in enumerate(z_sample):
    print("idx: {} value: {}".format(idx, value))
1
2
3
4
5

• Result

z sample: [[[-0.11212024 -0.69009897 -0.66992379]
  [-0.26245796 -0.86990319  0.00199974]
  [ 0.65475913 -0.61089172  0.37564042]]

 [[-0.54160346  0.5907193   0.9643114 ]
  [ 0.34624219  0.31073878 -0.78359222]
  [ 0.26297166 -0.70019566  0.0950346 ]]]
idx: 0 value: [[-0.11212024 -0.69009897 -0.66992379]
 [-0.26245796 -0.86990319  0.00199974]
 [ 0.65475913 -0.61089172  0.37564042]]
idx: 1 value: [[-0.54160346  0.5907193   0.9643114 ]
 [ 0.34624219  0.31073878 -0.78359222]
 [ 0.26297166 -0.70019566  0.0950346 ]]
1
2
3
4
5
6
7
8
9
10
11
12
13

• Analysis
  (1) 三维数组[2, 3, 3],enumerate的编号为第一维度2,取值为0,1,数据为[3, 3]的数组;
  (2) 对于图像处理,如Tensorflow的四维数组[1, 250, 250, 3]或batch图像[128, 250, 250, 3]可用于提取图像数据,用于保存或编辑;

3 sum

格式
sum(iterable, start)
功能
计算iterable(可遍历数据list，tuple，set)和，start为需要另外相加的数。

import numpy as np
a = np.array([1, 2, 3])
print("a: {}".format(a))
b = a.tolist()
print("b: {}".format(b))
c = sum(b)
print("sum: {}".format(c))
d = sum(b, 2)
print("sum extra add 2: {}".format(d))
1
2
3
4
5
6
7
8
9

• Result

a: [1 2 3]
b: [1, 2, 3]
sum: 6
sum extra add 2: 8
1
2
3
4

• Analysis
  实现列表中所有元素的求和，若需要指定相加的元素添加第二个参数，默认为0.

4 总结

(1) enumerate为数据提供编号,可通过编号处理相应数据;
(2) 二维或多维数组使用enumerate进行处理,编号为数组的第一个维度数据,数据为其他维度;
(3) 对于图像处理,如Tensorflow的四维数组[1, 250, 250, 3]或batch图像[128, 250, 250, 3],使用enumerate可用于提取图像数据,用于保存或编辑.

---

[参考文献]
[1]https://docs.python.org/3.6/library/functions.html?highlight=enumerate#enumerate

---