pyt*_*ict 5 python matplotlib pandas
我有一个csv文件,其中包含性别和婚姻状态以及更多列,如下所示.
Loan_ID,Gender,Married,Dependents,Education,Self_Employed,ApplicantIncome,CoapplicantIncome,LoanAmount,Loan_Amount_Term,Credit_History,Property_Area,Loan_Status
LP001002,Male,No,0,Graduate,No,5849,0,,360,1,Urban,Y
LP001003,Male,Yes,1,Graduate,No,4583,1508,128,360,1,Rural,N
LP001005,Male,Yes,0,Graduate,Yes,3000,0,66,360,1,Urban,Y
LP001006,Male,Yes,0,Not Graduate,No,2583,2358,120,360,1,Urban,Y
LP001008,Male,No,0,Graduate,No,6000,0,141,360,1,Urban,Y
LP001011,Male,Yes,2,Graduate,Yes,5417,4196,267,360,1,Urban,Y
Run Code Online (Sandbox Code Playgroud)
我想要数不.已婚男性和女性的结果如下图所示
以下是我使用的代码:
import csv
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
if __name__ == '__main__':
x=[]
y=[]
df = pd.read_csv(
"/home/train.csv",usecols=[1,2]).dropna(subset=['Gender','Married']) # Reading the dataset in a dataframe using Pandas
groups = df.groupby(['Gender','Married'])['Married'].apply(lambda x: x.count())
print(groups)
Run Code Online (Sandbox Code Playgroud)
分组后我有以下结果:
Gender Married
Female No 80
Yes 31
Male No 130
Yes 357
Run Code Online (Sandbox Code Playgroud)
我想要一个如下图表
jez*_*ael 14
您可以使用groupby+ size然后使用Series.plot.bar:
groups = df.groupby(['Gender','Married']).size()
groups.plot.bar()
Run Code Online (Sandbox Code Playgroud)
print (df.groupby(['Gender','Married']).size().unstack(fill_value=0))
Married No Yes
Gender
Female 80 31
Male 130 357
df.groupby(['Gender','Married']).size().unstack(fill_value=0).plot.bar()
Run Code Online (Sandbox Code Playgroud)
要么:
pd.crosstab(df['Gender'],df['Married']).plot.bar()
Run Code Online (Sandbox Code Playgroud)