Matplotlib 画红包

不会Photoshop,可以用Matplotlib画红包啊！

def PlotHb(i,vlist):    
    import matplotlib.pyplot as plt
    import matplotlib.transforms as mtransforms
    import matplotlib.patches as mpatch
    from matplotlib.patches import FancyBboxPatch
    import matplotlib
    
    matplotlib.rcParams['font.sans-serif'] = ['SimHei']
    matplotlib.rcParams['font.family']='sans-serif'


    fig = plt.figure(figsize=(25,10))
    ax = fig.add_subplot(111)
    rect = plt.Rectangle((0.0,0.0),0.35,1,fc='firebrick',edgecolor='yellow')


    rect2 = plt.Rectangle((0.6,0.0),0.35,1,fc='black',ec='yellow')
    ax.add_patch(rect)
    ax.add_patch(rect2)

    plt.plot((0,0.175),(0.95,0.65),color='yellow')
    plt.plot((0.175,0.35),(0.65,0.95),color='yellow')

    plt.plot((0.6,0.775),(0.95,0.65),color='yellow')
    plt.plot((0.775,0.95),(0.65,0.95),color='yellow')
    #plt.annotate()

    plt.axis('off')
    plt.text(0.45,0.5,"VS",fontsize=65)
    
    plt.text(0.01,0.3,vlist['A单位'] + str(" ")+str(vlist['A钱包']) + "元",fontsize=65)
    plt.title(i,fontsize=65)

    plt.show()


import pandas as pd
df=pd.read_csv('e:/xg.csv',encoding='gbk')
df.set_index('组名',inplace=True)
    
zm=df.index.tolist()
for i in zm:
    PlotHb(i,df.loc[i])