如何在两个不同的数据集上进行相同的置乱，但同时保存两个数据集中的行的顺序

import pandas as pd
import numpy as np

# 加载两个数据集
data1 = pd.read_csv('data1.csv')
data2 = pd.read_csv('data2.csv')

# 确保两个数据集具有相同的行数
assert len(data1) == len(data2)

# 创建索引列保存原始顺序
data1['index'] = np.arange(len(data1))
data2['index'] = np.arange(len(data2))

# 对两个数据集进行随机置乱
data1_shuffled = data1.sample(frac=1, random_state=42)
data2_shuffled = data2.sample(frac=1, random_state=42)

# 保存置乱后的数据集
data1_shuffled.to_csv('data1_shuffled.csv', index=False)
data2_shuffled.to_csv('data2_shuffled.csv', index=False)

# 如果需要恢复数据集的原始顺序，使用索引列排序
data1_restored = data1_shuffled.sort_values('index')
data2_restored = data2_shuffled.sort_values('index')