import pandas as pd #引入pandas包
data = pd.read_csv('data.csv') #读取数据文件
print(type(data))
print(data)
x y
0 -10 100
1 -9 81
2 -8 64
3 -7 49
4 -6 36
5 -5 25
6 -4 16
7 -3 9
8 -2 4
9 -1 1
10 0 0
11 1 1
12 2 4
13 3 9
14 4 16
15 5 25
16 6 36
17 7 49
18 8 64
19 9 81
20 10 100
x = data.loc[:,'x'] #打印x的列
y = data.loc[:,'y'] #打印x的列
print(y)
0 100
1 81
2 64
3 49
4 36
5 25
6 16
7 9
8 4
9 1
10 0
11 1
12 4
13 9
14 16
15 25
16 36
17 49
18 64
19 81
20 100
Name: y, dtype: int64
c = data.loc[:,'x'][y>50] #筛选数据y>50
print(c)
0 -10
1 -9
2 -8
18 8
19 9
20 10
Name: x, dtype: int64
data_array = np.array(data) #转换为数组
print(type(data_array))
print(data_array)
[[-10 100]
[ -9 81]
[ -8 64]
[ -7 49]
[ -6 36]
[ -5 25]
[ -4 16]
[ -3 9]
[ -2 4]
[ -1 1]
[ 0 0]
[ 1 1]
[ 2 4]
[ 3 9]
[ 4 16]
[ 5 25]
[ 6 36]
[ 7 49]
[ 8 64]
[ 9 81]
[ 10 100]]
data_new = data+10
#print(data_new)
data_new.head() #展现部分数据
# data to csv file
data_new.to_csv('data_new.csv') #保存数据文件
