python - 如何更改输出格式并将所有数字显示为列表？

Question

def printintersection(arr1,arr2,m,n):
    i,j=0,0
    while i<m and j<n:
        if arr1[i]<arr2[j]:
            i+=1
        elif arr2[j]<arr1[i]:
            j+=1
        else:
            print(arr2[j])
            j+=1
            i+=1
arr1 = [1, 2, 3, 4, 5]
arr2 = [3,4,5,6]
m=len(arr1)
n=len(arr2)
printintersection(arr1,arr2,m,n)

**

---

我得到如下所述的输出。

3
4
5

我想要列表格式的输出，即

[3,4,5]

提前致谢

Answer 1

只需对上述代码进行少量更改，您就可以通过将每个数字附加到列表并在末尾打印列表来获得所需的结果。

快照代码

def printintersection(arr1,arr2,m,n):
    l = []
    i,j=0,0
    while i<m and j<n:
        if arr1[i]<arr2[j]:
            i+=1
        elif arr2[j]<arr1[i]:
            j+=1
        else:
            l.append(arr2[j])
            j+=1
            i+=1
    print(l)

arr1 = [1, 2, 3, 4, 5]
arr2 = [3,4,5,6]
m=len(arr1)
n=len(arr2)
printintersection(arr1,arr2,m,n)                                                              

但我相信，列表理解是实现此目的的一种更简单的方法:

arr1 = [1, 2, 3, 4, 5]
arr2 = [3,4,5,6]
l = [num for num in arr1 if num in arr2]
print(l)