Я хочу подсчитать элементы списка python, который находится внутри фрейма данных, и чтобы результат был столбцом в фрейме данных. Как мне это сделать?

#python #pandas #dataframe #csv

#python #pandas #фрейме данных #csv

Вопрос:

Итак, у меня есть фрейм данных, который выглядит примерно так:

 | column_1 | column 2        |
| 0.       | [`a`, 'b', 'c'] |
| 1        | ['z', 'y']      |
| 2        | ['a']           |
 

И вывод, который я хочу, — это столбец (с теми же номерами индексов), который выглядит следующим образом:

 | column_1 | column 2        | column 3|
| 0.       | [`a`, 'b', 'c'] | 3       |
| 1        | ['z', 'y']      | 2       |
| 2        | ['a']           | 1       |
 

Ответ №1:

Просто сделайте:

 df['column_3'] =  df['column_2'].apply(len)
print(df)
 

Выходной сигнал

    column_1   column_2  column_3
0         0  [a, b, c]         3
1         1     [z, y]         2
2         2        [a]         1
 

Или немного более сложным, используйте str accesor:

 df['column_3'] = df['column_2'].str.len()
print(df)
 

Выходной сигнал

    column_1   column_2  column_3
0         0  [a, b, c]         3
1         1     [z, y]         2
2         2        [a]         1