在变量中打印最终列表

Printing final list in variable

我正在学习 Python 中的编程面试基础,我正在尝试为数组一章中的第一个问题找到替代解决方案。 这个想法是你要编写一个程序,它接受数组 A 和 A 中的索引 i 并重新排列元素,以便所有小于 A[i] 的元素首先出现,然后是等于主元的元素,然后是大于主元的元素枢。在书中,他们已经提供了一个解决方案,但我正在尝试找出一个替代方案。简而言之,我正在为每个方面创建子数组,例如小于、等于和大于 A[i]。目前,我正在努力存储小于主元的 A 的整数值。我想遍历 A 的列表,并将小于枢轴的所有元素的值存储在较小的(变量)中。这个想法是最终 return 变量更小,它将包含所有小于主元的值。为了检查我的工作,我使用了打印功能来检查更小的值。它在该变量中存储小于主元的每次迭代。理想情况下,使用这种方法我只想 return 较小变量的最终迭代而不是每次迭代。我的下一步应该是什么?希望这是有道理的,我真的不介意详细说明任何部分。提前致谢。

def properArray(pivot_index, A):
    pivot = A[pivot_index]
    smaller = []
    for i in range(len(A)):
        if A[i] < pivot:
            smaller.append(A[i])
            print (smaller)

resized_array =properArray
resized_array(3, [1,5,6,9,3,4,6])

我想这就是您要实现的目标

def properArray(pivot_index, A):
     pivot = A[pivot_index]
     smaller = []
     for i in range(len(A)):
         if A[i] < pivot:
             smaller.append(A[i])
     print (smaller)

resized_array =properArray
resized_array(3, [1,5,6,9,3,4,6])

不是每次迭代都打印 smaller 数组,而是需要在 for 循环完成后打印一次