Notice
Recent Posts
Recent Comments
Link
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | |||||
| 3 | 4 | 5 | 6 | 7 | 8 | 9 |
| 10 | 11 | 12 | 13 | 14 | 15 | 16 |
| 17 | 18 | 19 | 20 | 21 | 22 | 23 |
| 24 | 25 | 26 | 27 | 28 | 29 | 30 |
| 31 |
Tags
- 유니티
- MS-SQL
- 다이어트
- node.js
- 맛집
- mssql
- ASP
- MySQL
- swift
- IOS
- python
- urllib
- 날짜
- Excel
- flutter
- javascript
- 함수
- GIT
- 리눅스
- ubuntu
- port
- Linux
- PyQt5
- PER
- Unity
- 라즈베리파이
- sqlite
- tensorflow
- PyQt
- pandas
Archives
아미(아름다운미소)
Polars type 변경 본문
import polars as pl
def process_dataframe_optimized_pl(dict_df_types: dict, df: pl.DataFrame) -> pl.DataFrame:
def handle_column(col: str, dtype: str) -> pl.Expr:
try:
expr = pl.col(col)
if dtype == 'int':
return expr.cast(pl.Int32).fill_null(0).alias(col)
elif dtype == 'float':
return expr.cast(pl.Float32).fill_null(0).alias(col)
elif dtype == 'bool':
return expr.cast(pl.Utf8).str.to_lowercase().is_in(['true', 't', '1']).alias(col)
elif dtype == 'datetime':
return expr.cast(pl.Utf8).str.strptime(pl.Datetime, strict=False).alias(col)
elif dtype == 'string':
return expr.cast(pl.String).alias(col)
elif dtype == 'category':
return expr.cast(pl.Categorical).alias(col)
else:
return expr # return original if dtype not recognized
except Exception as e:
print(f"컬럼 '{col}' 처리 중 오류 발생: {e}")
return expr # return original on error
# Get intersection of DataFrame columns and dictionary keys
valid_cols = set(df.columns) & set(dict_df_types.keys())
# Filter for only valid types we want to process
valid_types = {'int', 'float', 'bool', 'datetime', 'string', 'category'}
# Create expressions for columns that need processing
exprs = [
handle_column(col, dict_df_types[col])
for col in valid_cols
if dict_df_types.get(col) in valid_types
]
return df.with_columns(exprs)
Comments