正文

python 对列表中任意两个数进行操作 (python operate any two elements in list)

IP云
IP云V管理员/5 阅读
09/10
提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

python中, 实现列表中的整型元素两两相乘或列表中的数组元素两两相与

1. 假设列表中的元素是整型, 可调用以下函数:

def list_any_two_mul(mylist):
     num = 1
     temp = []
     for i in mylist[:-1]:
         temp.append([i * j for j in mylist[num:]])
         num = num + 1
     # 把多个列表变成只有一个列表
     results = [y for x in temp for y in x]
     return results

假设有列表 mylist = [3, 5, 6, 8], 调用了以上的函数 list_any_two_mul(mylist) 则返回 [15, 18, 24, 30, 40, 48]

2. 假设列表中的元素是数组形式, 可以调用以下函数:

def list_any_two_and(mylist2):
     temp2 = []
     num2 = 1
     # 先对列表中的元素两两相与
     for i in mylist2[:-1]:
         temp2.append([cv2.bitwise_and(i, j) for j in mylist2[num2:]])
         num2 = num2 + 1
     # 再把相与后的结果放在一个列表中
     results2 = []
     for i in range(len(temp2)):
         for j in range(len(temp2[i])):
             results2.append(temp2[i][j])
     return results2

验证以上函数, 先生成元素是数组的列表,再调用函数, 具体实现可参照如下代码:

>>> import numpy as np
>>> a=np.arange(1,13).reshape(3,4)
>>> a
>>> array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])
>>> b=np.arange(13,25).reshape(3,4)
>>> b
>>> array([[13, 14, 15, 16],
       [17, 18, 19, 20],
       [21, 22, 23, 24]])
>>> c=np.random.randint(0,2,(3,4))
>>> c
>>> array([[0, 1, 1, 1],
       [0, 0, 1, 1],
       [0, 0, 1, 1]])
>>> d=[a,b,c]
>>> g=list_any_two_and(g)
>>> g
>>> [array([[1, 2, 3, 0],
        [1, 2, 3, 0],
        [1, 2, 3, 8]], dtype=int32), array([[0, 0, 1, 0],
        [0, 0, 1, 0],
        [0, 0, 1, 0]], dtype=int32), array([[0, 0, 1, 0],
        [0, 0, 1, 0],
        [0, 0, 1, 0]], dtype=int32)]
本人水平有限, 期待有更好更简便的方法, 有错误欢迎指正, 谢谢.