67 lines
3.2 KiB
Python
67 lines
3.2 KiB
Python
|
import pandas as pd
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
|
||
|
|
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
|
||
|
|
|
||
|
|
|
||
|
|
def read_csv_columns(csv_file_path, column_names):
|
||
|
|
df = pd.read_csv(csv_file_path)
|
||
|
|
return df[column_names].values
|
||
|
|
|
||
|
|
|
||
|
|
def plot_comparison(csv_file_path1, csv_file_path2, column_names):
|
||
|
|
data1 = read_csv_columns(csv_file_path1, column_names)
|
||
|
|
data2 = read_csv_columns(csv_file_path2, column_names)
|
||
|
|
|
||
|
|
plt.figure(figsize=(12, 8))
|
||
|
|
plt.plot(data1[:, 0], label=f'Person 1 - {column_names[0]}', color='#FF4081')
|
||
|
|
plt.plot(data2[:, 0], label=f'Person 2 - {column_names[0]}', linestyle='--', color='#FF4081')
|
||
|
|
plt.plot(data1[:, 1], label=f'Person 1 - {column_names[1]}', color='#1da1e6')
|
||
|
|
plt.plot(data2[:, 1], label=f'Person 2 - {column_names[1]}', linestyle='--', color='#1da1e6')
|
||
|
|
plt.plot(data1[:, 2], label=f'Person 1 - {column_names[2]}', color='#000210')
|
||
|
|
plt.plot(data2[:, 2], label=f'Person 2 - {column_names[2]}', linestyle='--', color='#000210')
|
||
|
|
plt.title('不同用户对比')
|
||
|
|
plt.xlabel('Time')
|
||
|
|
plt.ylabel('Value')
|
||
|
|
plt.legend(loc='lower left')
|
||
|
|
plt.tight_layout()
|
||
|
|
plt.savefig('ComparsionACC_Diff.png')
|
||
|
|
|
||
|
|
plt.figure(figsize=(12, 8))
|
||
|
|
plt.plot(data1[:, 3], label=f'Person 1 - {column_names[3]}', color='#FF4081')
|
||
|
|
plt.plot(data2[:, 3], label=f'Person 2 - {column_names[3]}', linestyle='--', color='#FF4081')
|
||
|
|
plt.plot(data1[:, 4], label=f'Person 1 - {column_names[4]}', color='#1da1e6')
|
||
|
|
plt.plot(data2[:, 4], label=f'Person 2 - {column_names[4]}', linestyle='--', color='#1da1e6')
|
||
|
|
plt.plot(data1[:, 5], label=f'Person 1 - {column_names[5]}', color='#000210')
|
||
|
|
plt.plot(data2[:, 5], label=f'Person 2 - {column_names[5]}', linestyle='--', color='#000210')
|
||
|
|
plt.title('不同用户对比')
|
||
|
|
plt.xlabel('Time')
|
||
|
|
plt.ylabel('Value')
|
||
|
|
plt.legend(loc='lower left')
|
||
|
|
plt.tight_layout()
|
||
|
|
plt.savefig('ComparsionGyro_Diff.png')
|
||
|
|
|
||
|
|
plt.figure(figsize=(12, 8))
|
||
|
|
plt.plot(data1[:, 6], label=f'Person 1 - {column_names[6]}', color='#FF4081')
|
||
|
|
plt.plot(data2[:, 6], label=f'Person 2 - {column_names[6]}', linestyle='--', color='#FF4081')
|
||
|
|
plt.plot(data1[:, 7], label=f'Person 1 - {column_names[7]}', color='#1da1e6')
|
||
|
|
plt.plot(data2[:, 7], label=f'Person 2 - {column_names[7]}', linestyle='--', color='#1da1e6')
|
||
|
|
plt.plot(data1[:, 8], label=f'Person 1 - {column_names[8]}', color='#000210')
|
||
|
|
plt.plot(data2[:, 8], label=f'Person 2 - {column_names[8]}', linestyle='--', color='#000210')
|
||
|
|
plt.title('不同用户对比')
|
||
|
|
plt.xlabel('Time')
|
||
|
|
plt.ylabel('Value')
|
||
|
|
plt.legend(loc='lower left')
|
||
|
|
plt.tight_layout()
|
||
|
|
plt.savefig('ComparsionMag_Diff.png')
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
csv_file1 = r'D:\pycharmProjects\python_server\NormalizeData\44c3071c5d754df686c47ce236428bf6_wavelet_denoised_kalman_denoised_normalized.csv'
|
||
|
|
csv_file2 = r'D:\pycharmProjects\python_server\NormalizeData\57edc01fed864a18b095fe7d5cf41ea2_wavelet_denoised_kalman_denoised_normalized.csv'
|
||
|
|
columns_to_compare = ['accX', 'accY', 'accZ', 'gyroX', 'gyroY', 'gyroZ', 'magX', 'magY', 'magZ']
|
||
|
|
|
||
|
|
plot_comparison(csv_file1, csv_file2, columns_to_compare)
|