import random
dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]
trainDataSet = random.sample(dataSet, 3)

import csv
import random
def getDataSet(proportion):
  """
    :exception
      獲取訓練集和測試集(將數據按比例隨機劃分)
    :parameter
      proportion - 測試集/數據集
    :return
      trainDataSet - 訓練集
      testDataSet - 測試集
    author
      肖政宇
    modify
      2019年5月10日
  """
  dataSet = open('數據集.csv')
  dataSetReader = csv.reader(dataSet)
  """
    :exception
      將數據保存到數組
  """
  dataSet = []
  next(dataSetReader, 'none') # 跳過表頭
  data = next(dataSetReader, 'none')
  while (data != 'none'):
    dataSet.append(data)
    data = next(dataSetReader, 'none')
  """
    :exception
      按照比例隨機劃分出訓練集和測試集
  """
  dataNumber = dataSet.__len__() # 數據集數據條數
  testNumber = int(dataNumber * proportion) # 測試集數據條數
  testDataSet = [] # 測試數據集
  trainDataSet = [] # 訓練數據集
 
  testDataSet = random.sample(dataSet, testNumber) # 測試集
  for testData in testDataSet: # 將已經選定的測試集數據從數據集中刪除
    dataSet.remove(testData)
  trainDataSet = dataSet # 訓練集
 
  return trainDataSet, testDataSet