Python进行数组合并的方法

Python013

Python进行数组合并的方法,第1张

python的数组合并在算法题中用到特别多,这里简单总结一下:

假设有a1和a2两个数组:

a1=[1,2,3]

a2=[4,5,6]

1. 直接相加

合并后赋值给新数组a3

a3 = a1 + a2

2. extend

调用此方法,a1会扩展成a1和a2的内容 a1.extend(a2)

3. 列表表达式

先生成新的二维数组) a3 = [a1, a2])

列表推导形成新的数组) a4 = [ y for a in a3 for y in a ])

下面分别测试下三种数组合并方式的性能

分别输出:

17.2916171551

20.8185400963

55.1758739948

可以看出:在数据量大的时候,第一种方式的性能要高出很多。

python里数列相邻数相加:

1、使用 for循环,定义一个累加求和函数sum2(n),for循环的作用就是循环遍历。def Sum(*args): count = 0 for i in args: count+=ireturn count

2、使用递归函数,定义一个累加求和函数sum3(n),递归函数一定要设置递归的出口,即当函数满足一个条件时,函数不再执行,目的防止出现死循环;设置当n=1时 ,我们让函数返回1,return后面的代码不在执行。def sum_numbers(num): # 1.出口 if num == 1: return 1 # 2.数组累加 temp = sum_numbers(num - 1) return num + temp result = sum_numbers(3)print(result)

这个如果用numpy我不怎么熟,我觉得用pandas是常态- -,代码如下:

import pandas as pd

a = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]

b=pd.DataFrame(a)

print(list(b.sum()))