Как получить первый столбец списка?

Question

Я хочу извлечь данные для первого столбца. Я хочу, чтобы на выходе было только [male, farmer], а остальные игнорируются. Что я должен делать? Спасибо за помощь! Пожалуйста, помогите, спасибо!

mode: [array(['male'], dtype='<U11'), array(['farmer'], dtype='<U11')]

Answer 1

Это помогает?

In [1]: l=[[1,2,3],[4,5,6],[7,8,9]]
In [2]: [sub[0] for sub in l]
Out[2]: [1, 4, 7]

Answer 2

>>> import numpy as np
>>> x = np.array([[1,2,3,4],[5,6,7,8]])

>>> x
array([[1, 2, 3, 4],
    [5, 6, 7, 8]])

>>> x[:,0] # for returning the first column using '0'
array([1, 5])

Я думаю, это может помочь вам

Answer 3

ОБНОВЛЕНИЕ

Попробуйте:

[element[0][0] for element in mode]

Как правило, [x для x в iterable_structure] позволит вам получить доступ к каждому элементу в списке, обработать его и создать новый список. Взгляните на:

https://docs.python.org/3/tutorial/datastructures.html#list-comprehensions