import numpy as np
import pandas as pd

id_list = []
num_list = []
train_list = {'id':[], 'num':[], 'camera':[]}
test_list = {'id':[], 'num':[], 'camera':[]}
more_list = {'id':[], 'num':[], 'camera':[]}

df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/test.xls', sheet_name='xiaomi13')
# 读取指定区域的两列数据
id_list = df['id'].tolist()
num_list = df['num'].tolist()

train_list['id'] = id_list
train_list['num'] = num_list
train_list['camera'] = ['mate30']

np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/test_xiaomi13.npy', train_list)

# df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/test.xls', sheet_name='mate30')
# # 读取指定区域的两列数据
# id_list = df['id'].tolist()
# num_list = df['num'].tolist()

# test_list['id'] = id_list
# test_list['num'] = num_list
# test_list['camera'] = ['mate30']

# np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/test_mate30.npy', test_list)

# contain = np.load('/home/wsy/cctcc-main/dataset/TCSR-Set/train_P30pro.npy', allow_pickle=True)
# dic = contain.item()
# print(dic['id'])

# df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/train.xls', sheet_name='all')
# # 读取指定区域的两列数据
# id_list = df['id'].tolist()
# num_list = df['num'].tolist()

# test_list['id'] = id_list
# test_list['num'] = num_list
# #test_list['camera'] = ['xiaomi13']

# np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/train_all.npy', test_list)

# df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/outdoor.xls', sheet_name='xiaomi13')
# # 读取指定区域的两列数据
# id_list = df['id'].tolist()
# num_list = df['num'].tolist()

# train_list['id'] = id_list
# train_list['num'] = num_list
# train_list['camera'] = ['xiaomi13']

# np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/outdoor_xiaomi13.npy', train_list)

# df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/indoor.xls', sheet_name='xiaomi13')
# # 读取指定区域的两列数据
# id_list = df['id'].tolist()
# num_list = df['num'].tolist()

# test_list['id'] = id_list
# test_list['num'] = num_list
# test_list['camera'] = ['xiaomi13']

# np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/indoor_xiaomi13.npy', test_list)

# df = pd.read_excel('/home/wsy/cctcc-main/dataset/TCSR-Set/transition.xls', sheet_name='vivo')
# # 读取指定区域的两列数据
# id_list = df['id'].tolist()
# num_list = df['num'].tolist()

# more_list['id'] = id_list
# more_list['num'] = num_list
# more_list['camera'] = ['vivo']

# np.save('/home/wsy/cctcc-main/dataset/TCSR-Set/transition_vivo.npy', more_list)