Как перевести операции DataFrame pandas в Polars в Python? ⇐ Python

[Anonymous]

Я пытаюсь преобразовать некоторые операции DataFrame pandas в Polars в Python, но сталкиваюсь с трудностями, особенно с операциями по строкам и сравнениями по элементам. Вот код pandas, с которым я работаю:

Код: Выделить всё

import pandas as pd

df_a = pd.DataFrame({
"feature1": [1, 2, 3],
"feature2": [7, 8, 9],
})

df_b = pd.DataFrame({
"feature1": [3, 8, 2],
"feature2": [7, 4, 9],
})

if selection_mode == 'option1':
max_values = df_a.max(axis=1)
selected_features = df_a.eq(max_values, axis=0)
final_result = selected_features.mul(df_b).sum(axis=1) / selected_features.sum(axis=1)

elif selection_mode == 'option2':
above_avg = df_a.ge(df_a.mean(axis=1), axis=0)
combined_df = above_avg.mul(df_a).mul(df_b)
sum_combined = combined_df.sum(axis=1)
sum_above_avg = above_avg.mul(df_a).sum(axis=1)
final_result = sum_combined / sum_above_avg

Будем очень признательны за любые рекомендации по переводу этого кода pandas в Polars!


Подробнее здесь: https://stackoverflow.com/questions/789 ... -in-python