Как вычесть данные между столбцами, которые имеют одинаковый субфикс ⇐ Python

[Anonymous]

У меня есть пример данных Data, который имеет тот же субфикс, что и 001, 002, 003 .

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

import pandas as pd
import numpy as np

branch_names = [f"Branch_{i}" for i in range(1, 11)]
date_1 = '20241231'
date_2 = '20250214'
date_3 = '20250220'

data = {
'Branch': branch_names,
date_1 + '_001': np.random.randint(60, 90, 10),
date_1 + '_002': np.random.randint(60, 90, 10),
date_1 + '_003': np.random.randint(60, 90, 10),
date_2 + '_001': np.random.randint(60, 90, 10),
date_2 + '_002': np.random.randint(60, 90, 10),
date_2 + '_003': np.random.randint(60, 90, 10),
date_3 + '_001': np.random.randint(60, 90, 10),
date_3 + '_002': np.random.randint(60, 90, 10),
date_3 + '_003': np.random.randint(60, 90, 10)
}

# Chuyển thành DataFrame
df = pd.DataFrame(data)

Теперь я хочу вычесть данные между столбцами, которые имеют тот же субфикс, что и ниже Принцип: < /p>

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

df['diff_1_001'] = df[date_3 + '_001'] - df[date_2 + '_001']
df['diff_2_001'] = df[date_3 + '_001'] - df[date_1 + '_001']
df['diff_1_002'] = df[date_3 + '_002'] - df[date_2 + '_002']
df['diff_2_002'] = df[date_3 + '_002'] - df[date_1 + '_002']
df['diff_1_003'] = df[date_3 + '_003'] - df[date_2 + '_003']
df['diff_2_003'] = df[date_3 + '_003'] - df[date_1 + '_003']
df

Как вы видите, у нас есть одинаковый 001, 002, 003 , но префикс отличается. Поэтому я не хочу жесткого кода 001, 002, 003 , но автоматически вычитайте его, как указано выше.


Подробнее здесь: https://stackoverflow.com/questions/794 ... ame-subfix