1
我需要將數據幀拆分爲2部分。例如，如果數據幀在下面被分割隨機基於Col1中兩個文件應當從每個類別1,2-包含樣品和3根據列的值將數據幀拆分爲兩個文件
Col1 col2
1 a
1 b
2 c
2 d
3 e
到目前爲止我能夠通過使用sklearn.cross_validation import train_test_split到數據轉換成所期望的比例分割。 但我無法弄清楚如何分割來從每個類別中提取樣本。 所有幫助將不勝感激。謝謝。