Построение фрейма данных с разными значениями шкалы в Python

Question

У меня есть следующий фрейм данных

df = pd.DataFrame({
'Date': [1930, 1931, 1932, 1933,1934],
'Income': [2300000, 5698907, 5976753, 6086762, 6577780],
'Age': [22, 45, 35, 40, 28],
'Weight': [0.01, 0.003, 0.04, 0.08, 0.07]
}) 

Каждая переменная имеет разные значения шкалы. Я хочу построить варибалы на одном графике, но из-за разницы в масштабах варибалов я могу видеть только линию дохода. Я построил с использованием

df.plot(figsize=(20,10), linewidth=5, fontsize = 20);

Я решил использовать шкалу на основе того, что нашел в Интернете, поэтому сделал следующее:

import pandas as pd
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
df = scaler.fit_transform(df)

Затем я попытался построить фрейм данных после масштабирования функции, и он дал следующую ошибку:

AttributeError: 'numpy.ndarray' object has no attribute 'plot'

Я не уверен, что делать дальше. Цель состоит в том, чтобы отобразить все переменные на одном графике.

Answer 1

Я считаю, что вам нужно создать новое DataFrame, потому как fit_transform возвращение 2d numpy array:

import pandas as pd
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()

df = pd.DataFrame(scaler.fit_transform(df), columns=df.columns, index=df.index)
df.plot(figsize=(20,10), linewidth=5, fontsize = 20)