在 Python 中如何向一个已排序的数组(列表) 中插入一个数呢
在 Python 中如何向一个 已排序的数组(列表) 中插入一个数呢?
方法有很多种,关键在于原来数组是什么样的排序,用到啥排序方法效率高,就用哪种。我们来练习其中的几种插入方法,另外也掌握下遍历数组的思路。
假如我原来的数组本身就很简单。如就是一个空数组,或者直接放到头部、或直接放尾部。
我们用insert()插入进去就好了
import bisect
y_num = []
num = 5
i_num= bisect.bisect_left(y_num, num)#NUM应该放的位置
y_num.insert(i_num,num)#有的位置之后,就放进去就好了,要指定是原列表,不能导致其他数组异常
print (y_num)
运行下看看
插入成功,为了方便,我们自定义个函数,后边调用就好了
def insert_num(sorted_y_num, n_num):
index = bisect.bisect_left(sorted_y_num, n_num)
sorted_y_num.insert(index, n_num)
return sorted_y_num
这样我们就少了很多代码,不用每次插入都写要去bisect_left()和insert(),只要有原列表和要插入的数,调用就好了。
请看下面代码:
#前面的空列表我就可以这么来用
y_num = []
num = 5
print(insert_num(y_num, num)) # 输出: [5]
#插入到头部
y_num = [2, 4, 6]
num = 1
print(insert_num(y_num, num)) # 输出: [1, 2, 4, 6]
#插入到尾部
y_num = [2, 4, 6]#关键一点是有序的表,不能很难找到头和尾部
num = 8
print(insert_num(y_num, num)) # 输出: [2, 4, 6, 8]
运行如下:
我们让它给出的数字比较任意的话,那就有可能在任何一个位置,我们来看下
y_num = [1, 3, 5, 7, 9]
num = 6
print(insert_num(y_num, num)) # 输出: [1, 3, 5, 6, 7, 9]
看下运行,是否正常
说到这,一般性的插入一个数完全没啥问题,前提是排序的规则要有,才好找到位置。
有很多时候,找位置要自己去按规则去找,而不用bisect_left()方法,我们该怎么做呢,那么你就分析遍历了,
def insert_num(ynum_arr, new_num):
# 遍历找到插入位置
for i in range(len(ynum_arr)):#用原数组的长度
if ynum_arr[i] >= new_num:#和新的比较
break
else:
i = len(ynum_arr) # 如果目标数比所有元素大,插入末尾
# 插入目标数
ynum_arr.insert(i, new_num)
return ynum_arr
arr = [1, 3, 5, 7, 9]
num = 6
print(insert_num(arr, num)) # 输出: [1, 3, 5, 6, 7, 9]
运行下看看,
主要是遍历的时候变化大,比如二分查找方法等等。不同给数据结果,使用的方法区别很大,多琢磨。