Создание условных выражений
Преобразование данных с помощью Polars
Liam Brannigan
Data Scientist & Polars Contributor
Категоризация по качеству рейтинга
shape: (5, 5)
| business | location | type | rating | capacity |
| --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 |
|------------------|-----------------|------------|--------|----------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 |
| Bang Bang Burger | Forest Rd. | restaurant | 4 | 55 |
| Costa Coffee | City Point | café | 5 | 41 |
| Costa Coffee | The Moorgate | takeaway | 3 | 0 |
| The Queens Head | Denman St. | bar | 5 | 187 |
- Задача: выявить рестораны с высоким рейтингом
Категоризация по качеству рейтинга
ratings.with_columns(
)
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when( )
)
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when(pl.col("rating") == 5)
)
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when(pl.col("rating") == 5).then(pl.lit("Highly Rated"))
)
$$
$$
- Используйте
pl.lit(), чтобы передать литеральное значение, а не имя столбца
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when(pl.col("rating") == 5).then(pl.lit("Highly Rated"))
.otherwise(pl.lit("Needs Improvement"))
)
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when(pl.col("rating") == 5).then(pl.lit("Highly Rated"))
.otherwise(pl.lit("Needs Improvement")).alias("quality")
)
Категоризация по качеству рейтинга
ratings.with_columns(
pl.when(pl.col("rating") == 5).then(pl.lit("Highly Rated"))
.otherwise(pl.lit("Needs Improvement")).alias("quality")
)
shape: (5, 6)
| business | location | type | rating | capacity | quality |
| --- | --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 | str |
|------------------|-----------------|------------|--------|----------|--------------------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 | Highly Rated |
| Bang Bang Burger | Forest Rd. | restaurant | 4 | 55 | Needs Improvement |
| ... | ... | ... | ... | ... | ... |
Категоризация по размеру заведения
shape: (5, 5)
| business | location | type | rating | capacity |
| --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 |
|------------------|-----------------|------------|--------|----------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 |
| Bang Bang Burger | Forest Rd. | restaurant | 4 | 55 |
| Costa Coffee | City Point | café | 5 | 41 |
| Costa Coffee | The Moorgate | takeaway | 3 | 0 |
| The Queens Head | Denman St. | bar | 5 | 187 |
- Задача: отфильтровать рестораны по вместимости
Категоризация по размеру заведения
ratings.with_columns(
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100)
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100).then(pl.lit("Large"))
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100).then(pl.lit("Large"))
.when(pl.col("capacity") >= 20)
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100).then(pl.lit("Large"))
.when(pl.col("capacity") >= 20).then(pl.lit("Medium"))
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100).then(pl.lit("Large"))
.when(pl.col("capacity") >= 20).then(pl.lit("Medium"))
.otherwise(pl.lit("Small"))
)
Категоризация по размеру заведения
ratings.with_columns(
pl.when(pl.col("capacity") > 100).then(pl.lit("Large"))
.when(pl.col("capacity") >= 20).then(pl.lit("Medium"))
.otherwise(pl.lit("Small")).alias("venue_size")
)
shape: (3, 6)
| business | location | type | rating | capacity | venue_size |
| --- | --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 | str |
|------------------|-----------------|------------|--------|----------|------------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 | Medium |
| Costa Coffee | The Moorgate | takeaway | 3 | 0 | Small |
| The Queens Head | Denman St. | bar | 5 | 187 | Large |
Стандартизация типов заведений
shape: (5, 5)
| business | location | type | rating | capacity |
| --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 |
|------------------|-----------------|------------|--------|----------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 |
| Bang Bang Burger | Forest Rd. | restaurant | 4 | 55 |
| Costa Coffee | City Point | café | 5 | 41 |
| Costa Coffee | The Moorgate | takeaway | 3 | 0 |
| The Queens Head | Denman St. | bar | 5 | 187 |
- Задача: стандартизировать столбец
type
Стандартизация типов заведений
ratings.with_columns(
pl.col("type")
)
Стандартизация типов заведений
ratings.with_columns(
pl.col("type").replace( )
)
Стандартизация типов заведений
ratings.with_columns(
pl.col("type").replace("café", "restaurant")
)
shape: (5, 5)
| business | location | type | rating | capacity |
| --- | --- | --- | --- | --- |
| str | str | str | i64 | i64 |
|------------------|-----------------|------------|--------|----------|
| 7burgers | Wakey Wakey | restaurant | 5 | 55 |
| Bang Bang Burger | Forest Rd. | restaurant | 4 | 55 |
| Costa Coffee | City Point | restaurant | 5 | 41 |
| Costa Coffee | The Moorgate | takeaway | 3 | 0 |
| The Queens Head | Denman St. | bar | 5 | 187 |
Давайте потренируемся!
Преобразование данных с помощью Polars
