使用pandas读写csv文件

• 项目结构
  F:\PYCHARM\KAGGLE\DIGITRECOGNIZER
  │ DataProcessing.py
  │ main.py
  │ Model.py│
  └─data
  │ LinearClassifier.csv
  │ sample_submission.csv
  │ test.csv
  │ train.csv

train.csv

test.csv

读取csv

train = pd.read_csv("data/train.csv")

print(train.shape)
(3,2)

print(train)
x y
0 1 1
1 2 4
2 3 9

表头信息

print(train.info())

<class ‘pandas.core.frame.DataFrame’>
RangeIndex: 3 entries, 0 to 2
Data columns (total 2 columns):
# Column Non-Null Count Dtype
— ------ -------------- -----
0 x 3 non-null int64
1 y 3 non-null int64
dtypes: int64(2)
memory usage: 176.0 bytes

print(train.train.describe())

转换为数组

print(train.values)

[[1 1]
[2 4]
[3 9]]

选择某一列

print(train["y"])

0 1
1 4
2 9
Name: y, dtype: int64

删除某一列

train.drop("y",1)

x
0 1
1 2
2 3

导出csv

ID和Value是表头

result = list(range(1, 5))
submission = pd.DataFrame({"ID": range(1, 5), "Value": result})
submission.to_csv("data/result1.csv", index=False)
submission.to_csv("data/result2.csv", index=True)

result1.csv

result = list(range(1, 5))
submission = pd.DataFrame({"ID": range(1, 5), "Value": result})
submission.to_csv("data/result2.csv", index=True)
# 多了第一列的数字

result2.csv