Pandas: Thêm cột mới vào khung dữ liệu là bản sao của cột chỉ mục

Question

Tôi có một khung dữ liệu mà tôi muốn vẽ bằng matplotlib, nhưng cột chỉ mục là thời gian và tôi không thể vẽ nó.

Đây là dataframe (df3):

nhưng khi tôi thử như sau:

plt.plot(df3['magnetic_mag mean'], df3['YYYY-MO-DD HH-MI-SS_SSS'], label='FDI') 

Tôi nhận được một lỗi rõ ràng:

KeyError: 'YYYY-MO-DD HH-MI-SS_SSS' 

Vì vậy, những gì tôi muốn làm là thêm một cột mới vào dataframe của tôi (tên là 'Thời gian) mà tôi chỉ là một bản sao của cột chỉ mục.

Tôi có thể làm như thế nào?

Đây là toàn bộ mã:

#Importing the csv file into df 
df = pd.read_csv('university2.csv', sep=";", skiprows=1) 

#Changing datetime 
df['YYYY-MO-DD HH-MI-SS_SSS'] = pd.to_datetime(df['YYYY-MO-DD HH-MI-SS_SSS'], 
               format='%Y-%m-%d %H:%M:%S:%f') 

#Set index from column 
df = df.set_index('YYYY-MO-DD HH-MI-SS_SSS') 

#Add Magnetic Magnitude Column 
df['magnetic_mag'] = np.sqrt(df['MAGNETIC FIELD X (μT)']**2 + df['MAGNETIC FIELD Y (μT)']**2 + df['MAGNETIC FIELD Z (μT)']**2) 

#Subtract Earth's Average Magnetic Field from 'magnetic_mag' 
df['magnetic_mag'] = df['magnetic_mag'] - 30 

#Copy interesting values 
df2 = df[[ 'ATMOSPHERIC PRESSURE (hPa)', 
      'TEMPERATURE (C)', 'magnetic_mag']].copy() 

#Hourly Average and Standard Deviation for interesting values 
df3 = df2.resample('H').agg(['mean','std']) 
df3.columns = [' '.join(col) for col in df3.columns] 

df3.reset_index() 
plt.plot(df3['magnetic_mag mean'], df3['YYYY-MO-DD HH-MI-SS_SSS'], label='FDI') 

Cảm ơn bạn !!

Answer 1

Tôi nghĩ bạn cần reset_index.

df3.reset_index(inplace=True) 

Hoặc:

df3 = df3.reset_index() 

Nhưng nếu bạn cần cột mới, sử dụng:

df3['new'] = df3.index 

Tôi nghĩ rằng bạn có thể read_csv tốt hơn:

df = pd.read_csv('university2.csv', 
       sep=";", 
       skiprows=1, 
       index_col='YYYY-MO-DD HH-MI-SS_SSS', 
       parse_dates='YYYY-MO-DD HH-MI-SS_SSS') #if doesnt work, use pd.to_datetime 

Và sau đó bỏ qua:

#Changing datetime 
df['YYYY-MO-DD HH-MI-SS_SSS'] = pd.to_datetime(df['YYYY-MO-DD HH-MI-SS_SSS'], 
               format='%Y-%m-%d %H:%M:%S:%f') 
#Set index from column 
df = df.set_index('YYYY-MO-DD HH-MI-SS_SSS') 

Answer 2

Bạn có thể truy cập trực tiếp trong chỉ mục và làm cho nó âm mưu, sau đây là một ví dụ:

import matplotlib.pyplot as plt 
import pandas as pd 
import numpy as np 

df = pd.DataFrame(np.random.randn(1000), index=pd.date_range('1/1/2000', periods=1000)) 

#Get index in horizontal axis 
plt.plot(df.index, df[0]) 
plt.show() 

#Get index in vertiacal axis 
plt.plot(df[0], df.index) 
plt.show()