python - リストから特定の列を抽出する

Question

次のような列インデックスを含むリストがあります。

list1 = [0 ,2]

リストの別のリストには、次のようにcsvファイルのファイルの内容が含まれます。

list2=[["abc", 1, "def"], ["ghi", 2, "wxy"]]

ielist2に含まれている列番号の値のみを含む新しいリストを作成するための最良の方法は何でしょうか。list1

newList = [["abc", "def"], ["ghi", "wxy"]]

サブリストを作成するのに苦労しています

Answer 1

あなたが使用することができますList Comprehension: -

newList = [[each_list[i] for i in list1] for each_list in list2]

Answer 2

タプルのリストに満足している場合は、次を使用できますoperator.itemgetter

import operator
list1 = [0,2]
my_items = operator.itemgetter(*list1)
new_list = [ my_items(x) for x in list2 ]

（またはここで使用できmapます）：

new_list = map(my_items, list2)

そして1ライナーとして：

new_list = map(operator.itemgetter(*list1), list2)

operator.itemgetterネストされたリスト内包表記よりもパフォーマンスがわずかに優れている可能性がありますが、心配する価値がないほど小さい可能性があります。

Answer 3

>>> list1 = [0 ,2]
>>> list2=[["abc", 1, "def"], ["ghi", 2, "wxy"]]
>>> newList = [[l[i] for i in list1] for l in list2]
>>> print newList
[['abc', 'def'], ['ghi', 'wxy']]

Answer 4

csv ファイルで作業している場合は、一からやり直す必要はありません。優れたcsvモジュールをご覧ください。