從python中的列表列表中選擇一個子列表

Question

我有一個列表A的列表：

A = [ [0, 42, 2.6],
      [1, 20, 5.6],
      [2, 67, 3.5],
      [3, 12, 3.2],
      [4, 34, 1.1],
      [5, 74, 4.7],
      [6, 29, 2.9] ]

我想從中提取一個子列表B ，該子列表B僅包含第二列小於30的那些列表（即B = [[1, 20, 5.6],[3, 12, 3.2],[6, 29, 2.9]] ）

當然，我可以將其轉換為numpy並使用np.where 。 但是我想知道是否只有通過使用列表才能做到這一點。 我嘗試了列表理解，但是在這方面我不是很熟練：

B = [x for x in len(A) if x[1] <= 30]

但這確實不起作用。 有什么建議嗎？ 提前致謝。

Answer 1

嘗試這個：

B = [x for x in A if x[1] <= 30]

Answer 2

您當前的代碼正在嘗試生成一個值，該值將在每次迭代時用作索引。 要實現該方法，請使用x訪問A每一行：

B = [A[x] for x in range(len(A)) if A[x][1] <= 30]

但是請注意，使用列表推導或更實用的filter要簡單得多：

清單理解：

B = [i for i in A if i[1] <= 30]

filter ：

B = list(filter(lambda x:x[1] <= 30, A))

Answer 3

這是一種向量化方法：

import numpy as np

A = np.array([[0, 42, 2.6],
             [1, 20, 5.6],
             [2, 67, 3.5],
             [3, 12, 3.2],
             [4, 34, 1.1],
             [5, 74, 4.7],
             [6, 29, 2.9]])

A[A[:, 1] <= 30]

# array([[  1. ,  20. ,   5.6],
#        [  3. ,  12. ,   3.2],
#        [  6. ,  29. ,   2.9]])