如何從列表中刪除子列表？

Question

我有一份包含50,000個樣本的training_data清單。 每個樣本都包含兩個子列表，其中包含兩個元素（樣本，標簽）。

我不小心在標簽項目內創建了另一個列表，該列表又有兩個項目。

即

                              main_list

                           _______|_______
                          |               |
                       sample           label(list)
                                     _____|_____
                                a(list)        b(list)

我想從每個示例中刪除“ b”子列表（50000）。 希望我能正確解釋。 救我。

Answer 1

如果我理解正確，請嘗試new_list = [ [x[0], x[1][0]] for x in main_list] 。 無論如何，列表理解可能足以解決您的問題...

Answer 2

我試圖重新創建您的列表層次結構，如下所示：

a = ["w", "x"]
b = ["y", "z"]
label = [a, b]
sample = ["sample1", "sample2"]
main_list = [sample, label]

這給了我們

print(main_list)
[['sample1', 'sample2'], [['w', 'x'], [ 'y', 'z']]]

然后要刪除b ，只需設置

main_list = [main_list[0], main_list[1][0]] 

這樣能達到您想要的嗎？