根据列的索引值从 CSV 文件中拆分数据

Question

我正在使用一个CSV文件，其中包含从3种类型的信号(蓝牙，WiFi和无人机(中提取的特征。我想保存 3 个单独的数据帧，每个数据帧仅包含一种信号类型的特征。

url = 'https://raw.githubusercontent.com/vmatei2/Drone_detection/master/Features.csv'
names = ['I_std','I_mean','I_mad','Q_std', 'Q_mean' ,'Q_mad','I_kurtosis','Q_kurtosis','Signal_type']
data = pd.read_csv(url, names = names)

提前谢谢你！

Answer 1

如何使用groupby

gp = data.groupby('Signal_type').groups
bt_data = data.iloc[gp['BT']] 
wifi_data = data.iloc[gp['WIFI']]
drone_data = data.iloc[gp['Drone']]

或者像这样：

wifi_data = data[data['Signal_type'] == 'WIFI']