在變量中打印最終列表

Question

我正在通過 Python 中的編程面試元素工作，我正在嘗試為數組一章中的第一個問題找到替代解決方案。 這個想法是你要編寫一個程序，它接受一個數組 A 和索引 i 到 A 並重新排列元素，使得所有小於 A[i] 的元素首先出現，然后是等於樞軸的元素，然后是大於樞。 在書中，他們已經提供了解決方案，但我正在嘗試找出替代方案。 簡而言之，我正在為每個方面創建子數組，例如小於、等於和大於 A[i]。 現在我正在努力存儲小於樞軸的 A 的整數值。 我想遍歷 A 的列表並將小於樞軸的所有元素的值存儲在較小的（變量）中。 這個想法是最終返回更小的變量，該變量將包含所有小於樞軸的值。 為了檢查我的工作，我使用了打印功能來檢查較小的值。 它存儲的每次迭代都小於該變量中的樞軸。 理想情況下，使用這種方法我只想返回較小變量的最終迭代而不是每次迭代。 我的下一步應該是什么？ 希望這是有道理的，我真的不介意詳細說明任何部分。 提前致謝。

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])

Answer 1

我想這就是你想要達到的

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 循環完成后打印一次