Извлечь столбец из списка строк с Python

Question

У меня есть следующий набор данных, который представляет собой ряд строк хранится в виде вложенных списков:

[['John', '35', 'UK'], 
['Emma', '43', 'UK'], 
['Lucy', '25', 'AU']] 

(строки всегда такой же длины)

I необходимо вернуть 'UK', 'AU' в качестве итерабельного (безразлично к заказу).

Есть ли один вкладыш, который возвращает уникальные значения, содержащиеся в третьем столбце, и который проще этого?

set(list(map(list, zip(*l)))[2]) 

(Ref: Transpose list of lists)

Answer 1

Изменение вас собственный код:

Python 3.x:

set(list(zip(*l)[2])) 

Python 2.x:

set(zip(*l)[2]) 

Демо:

l=[['John', '35', 'UK'],['Emma', '43', 'UK'],['Lucy', '25', 'AU']] 
set(list(zip(*l)[2])) 
{'AU', 'UK'} 

Answer 2

>>> l = [['John', '35', 'UK'], 
     ['Emma', '43', 'UK'], 
     ['Lucy', '25', 'AU']] 
>>> set(element[-1] for element in l) 
('AU', 'UK') 

Answer 3

Вы можете использовать NumPy:

import numpy as np 

arr = np.array([['John', '35', 'UK'], 
       ['Emma', '43', 'UK'], 
       ['Lucy', '25', 'AU']]) 

unique = np.unique(arr[:,2]) 

Answer 4

Вы можете использовать list comprehension:

>>> L = [['John', '35', 'UK'], 
['Emma', '43', 'UK'], 
['Lucy', '25', 'AU']] 
>>> set([i[2] for i in L]) 
set(['AU', 'UK']) 

Answer 5

Я думаю, что на самом деле требования bsuire являются более сложными на практике , поэтому я рекомендую использовать pandas для обработки таких требований, он более мощный и гибкий.

так, как использовать панд в этом случае:

In [17]: import pandas as pd 

In [18]: a = [['John', '35', 'UK'], 
    ....: ['Emma', '43', 'UK'], 
    ....: ['Lucy', '25', 'AU']] 

In [19]: b = pd.DataFrame(a) 

In [20]: b 
Out[20]: 
     0 1 2 
0 John 35 UK 
1 Emma 43 UK 
2 Lucy 25 AU 

In [21]: b[2].unique() 
Out[21]: array(['UK', 'AU'], dtype=object) 

In [22]: