import numpy as np
import pandas as pd
#数据预处理与数据提取

#创建数据表1
df1 = pd.DataFrame({"id":[1001,1002,1003,1004,1005,1006], 
 "date":pd.date_range('20220102', periods=6),
  "city":['Beijing ', 'Hangzhou', 'Shanghai', 'Beijing', 'Shanghai', 'Shanghai'],
 "age":[31,27,25,27,29,28],
 "category":['100-B','100-B','110-A','110-C','210-C','130-F'],
  "price":[1000,np.nan,2300,5400,np.nan,3330]},
  columns =['id','date','city','category','age','price'])
  
#创建数据字典
col={'id':[1001,1002,1003,1004,1005,1006,1007,1008,1009],
    'gender':['F','M','F','F','M','M','M','F','F'],
    'name':['Jane','Wan','Summer','Flore','Wang','Chung','Dev','Linda','Lucy',],
    'salary':[3000,5500,3500,4600,4500,6000,3800,6500,3500]}
#创建数据表2
df2 = pd.DataFrame(col,
                columns =['id','gender','name','salary'])

#数据表合并
df_inner=pd.merge(df1,df2,how='inner')  # 匹配合并，交集
df_left=pd.merge(df1,df2,how='left')     
df_right=pd.merge(df1,df2,how='right')
df_outer=pd.merge(df1,df2,how='outer')  #并集

print(df_inner)
print(df_left)
print(df_right)
print(df_outer)

#添加数据，相当于union，和全连接full join
# df_addtion = df1.append(df2)  #把df2添加到df1后面
# print(df_addtion)

#df1和df2两个表进行连接：与append追加的结果一样
df_concat = pd.concat([df1,df2], axis=0, join='outer', ignore_index=False,
            keys=None, levels=None, names=None, verify_integrity=False,
            sort=False, copy=True)
#df1的id字段与df2的name字段进行连接
df_concat = pd.concat([df1.id,df2.name], axis=0, join='outer', ignore_index=False,
            keys=None, levels=None, names=None, verify_integrity=False,
            sort=False, copy=True)
print(df_concat)

#设置索引列
df1.set_index('id')  #把id列设置为索引列

#按索引提取单行的数值
df_inner.loc[5]  #提取“索引值”为5的那一行数据，也就是第6行

#按索引提取区域行数值
df_inner.iloc[0:3]  #用切片提取前3行

#重设索引
df_inner.reset_index()  #再设置一条索引

df_inner.loc[:4]  #因为是索引名，所以这里是取0到4的所有列
df_inner.loc[2:4]  #2到4的所有列

df_inner.loc[2:4,'age':'name']  #2到4的age到name列
df_inner.loc[:5,"city":"salary"]

df_inner.loc['2022-01-02':]  #提取2022年1月2日以后的所有行
df_inner.loc[:'2022-01-05']  #提取2022年1月5日之前的所有数据

df_inner.loc['2022-01-02':,'age':'name']  #提取2022年1月2日后的age到name列
df_inner.loc[:'2022-01-05',:]  #提取2022年1月5日前的所有列

#将索引定义为id，并且因为下面还需要用到替换后的表，因此选择替换源数据
df_inner.set_index('id',inplace=True)  
#把列名city替换为城市
df_inner.rename(columns={'city':'城市'},inplace=True)

#尝试：
df_inner.loc[1002:1005]  #id为1002-1005的信息
df_inner.loc[:,'城市':'name']  #城市‘列到’name‘列信息

#实现：id为1002-1005的’城市‘列到’name‘列信息
df_inner.loc[1002:1005,'城市':'name']

#使用iloc按位置区域提取数据
df_inner.iloc[:4,:5]  #从0开始，前四行，前五列
df_inner.iloc[1:3,2:]  #从1到3行，第2列以后的所有列

df_inner.iloc[[0,2,5],[4,5]]   #提取第0、2、4行，4、5列

#按照索引列排序
 df1.sort_index(ascending=False)  #ascending为False表示降序，默认升序
 
#按照特定列值排序
df2.sort_values(by='salary',ascending=False)  #按照salary工资从高到低排序

#单条件标记：工资大于3000为high，低于3000为low
df_inner['level'] = np.where(df_inner['salary'] > 3000,'high','low')  #df_inner是两表进行内连接的结果
print(df_inner)

#多条件分组标记：city为上海，工资大于4000等级为1
df_inner.loc[(df_inner['city'] == 'Shanghai') & (df_inner['salary'] >= 4000), 'level']=1
print(df_inner)

pd.DataFrame((x.split('-') for x in df_inner['category']),index=df_inner.id, columns=['category','size'])

df3=pd.DataFrame(df_inner['name'].str[:3])
df3.set_index('name')
print(df3)

#判断city列的值是否为北京
df_inner['city'].isin(['Shanghai'])

#判断city列里是否包含Beijing和Shanghai，然后将符合条件的数据提取出来
df_inner.loc[df_inner['city'].isin(['Beijing','Shanghai'])]