У меня есть файл df, который я хотел бы сгруппировать и записать в формат csv. Однако один из столбцов имеет тип списка, который предотвращает запись df в csv.
import polars as pl
df = pl.DataFrame({"Column A": ["Variable 1", "Variable 2", "Variable 2", "Variable 3", "Variable 3", "Variable 4"],
"Column B": ["AB", "AB", "CD", "AB", "CD", "CD"]})
Что я хочу сгруппировать, как показано ниже:
df.group_by("Column A").agg(pl.col("Column B").unique())
Выход:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 4 ┆ ["CD"] │
│ Variable 2 ┆ ["CD", "AB"] │
│ Variable 3 ┆ ["AB", "CD"] │
│ Variable 1 ┆ ["AB"] │
└────────────┴──────────────┘
При попытке записать приведенный выше фрейм данных в csv возникает ошибка:
# ComputeError: CSV format does not support nested data. Consider using a different data format. Got: 'list[str]'
При попытке преобразовать тип списка в pl.String возникает ошибка
(df
.group_by("Column A").agg(pl.col("Column B").unique())
.with_columns(pl.col("Column B").cast(pl.String))
)
Выход:
# InvalidOperationError: cannot cast List type (inner: 'String', to: 'String')
Если я попытаюсь разобрать список в контексте group_by:
df.group_by("Column A").agg(pl.col("Column B").unique().explode())
Вывод нежелателен:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 1 ┆ ["AB"] │
│ Variable 2 ┆ ["CD", "AB"] │
│ Variable 3 ┆ ["CD", "AB"] │
│ Variable 4 ┆ ["CD"] │
└────────────┴──────────────┘
Как мне удобнее всего сгруппировать_by, а затем записать в CSV?
Желаемый вывод, записанный в CSV:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 3 ┆ ["AB", "CD"] │
│ Variable 1 ┆ ["AB"] │
│ Variable 4 ┆ ["CD"] │
│ Variable 2 ┆ ["CD", "AB"] │
└────────────┴──────────────┘
Подробнее здесь: https://stackoverflow.com/questions/752 ... ted-string
Тип списка полярностей в строку, разделенную запятыми ⇐ Python
Программы на Python
-
Anonymous
1762095693
Anonymous
У меня есть файл df, который я хотел бы сгруппировать и записать в формат csv. Однако один из столбцов имеет тип списка, который предотвращает запись df в csv.
import polars as pl
df = pl.DataFrame({"Column A": ["Variable 1", "Variable 2", "Variable 2", "Variable 3", "Variable 3", "Variable 4"],
"Column B": ["AB", "AB", "CD", "AB", "CD", "CD"]})
Что я хочу сгруппировать, как показано ниже:
df.group_by("Column A").agg(pl.col("Column B").unique())
Выход:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 4 ┆ ["CD"] │
│ Variable 2 ┆ ["CD", "AB"] │
│ Variable 3 ┆ ["AB", "CD"] │
│ Variable 1 ┆ ["AB"] │
└────────────┴──────────────┘
При попытке записать приведенный выше фрейм данных в csv возникает ошибка:
# ComputeError: CSV format does not support nested data. Consider using a different data format. Got: 'list[str]'
При попытке преобразовать тип списка в pl.String возникает ошибка
(df
.group_by("Column A").agg(pl.col("Column B").unique())
.with_columns(pl.col("Column B").cast(pl.String))
)
Выход:
# InvalidOperationError: cannot cast List type (inner: 'String', to: 'String')
Если я попытаюсь разобрать список в контексте group_by:
df.group_by("Column A").agg(pl.col("Column B").unique().explode())
Вывод нежелателен:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 1 ┆ ["AB"] │
│ Variable 2 ┆ ["CD", "AB"] │
│ Variable 3 ┆ ["CD", "AB"] │
│ Variable 4 ┆ ["CD"] │
└────────────┴──────────────┘
Как мне удобнее всего сгруппировать_by, а затем записать в CSV?
Желаемый вывод, записанный в CSV:
shape: (4, 2)
┌────────────┬──────────────┐
│ Column A ┆ Column B │
│ --- ┆ --- │
│ str ┆ list[str] │
╞════════════╪══════════════╡
│ Variable 3 ┆ ["AB", "CD"] │
│ Variable 1 ┆ ["AB"] │
│ Variable 4 ┆ ["CD"] │
│ Variable 2 ┆ ["CD", "AB"] │
└────────────┴──────────────┘
Подробнее здесь: [url]https://stackoverflow.com/questions/75221856/polars-list-type-to-comma-separated-string[/url]
Ответить
1 сообщение
• Страница 1 из 1
Перейти
- Кемерово-IT
- ↳ Javascript
- ↳ C#
- ↳ JAVA
- ↳ Elasticsearch aggregation
- ↳ Python
- ↳ Php
- ↳ Android
- ↳ Html
- ↳ Jquery
- ↳ C++
- ↳ IOS
- ↳ CSS
- ↳ Excel
- ↳ Linux
- ↳ Apache
- ↳ MySql
- Детский мир
- Для души
- ↳ Музыкальные инструменты даром
- ↳ Печатная продукция даром
- Внешняя красота и здоровье
- ↳ Одежда и обувь для взрослых даром
- ↳ Товары для здоровья
- ↳ Физкультура и спорт
- Техника - даром!
- ↳ Автомобилистам
- ↳ Компьютерная техника
- ↳ Плиты: газовые и электрические
- ↳ Холодильники
- ↳ Стиральные машины
- ↳ Телевизоры
- ↳ Телефоны, смартфоны, плашеты
- ↳ Швейные машинки
- ↳ Прочая электроника и техника
- ↳ Фототехника
- Ремонт и интерьер
- ↳ Стройматериалы, инструмент
- ↳ Мебель и предметы интерьера даром
- ↳ Cантехника
- Другие темы
- ↳ Разное даром
- ↳ Давай меняться!
- ↳ Отдам\возьму за копеечку
- ↳ Работа и подработка в Кемерове
- ↳ Давай с тобой поговорим...
Мобильная версия