获取DataFrame第一列的值(无魔法数字)

#encoding:utf-8
import pandas as pd
import matplotlib.pyplot as plt

#最美的时候遇见你 少年班 既然青春留不住
cols='name add remove gongsi daoyan yanyuan type boxoffice area'.split()
data=pd.read_csv('../data/film_log3.csv',names=cols,sep=';')
data['add']=pd.to_datetime(data['add'])
data['remove']=pd.to_datetime(data['remove'])
def get_data(_df):
    try:
        _box=_df['boxoffice'][15:]
        _df['boxoffice']=float(_box)
    except:
        pass
    try:
        _day=(_df['remove']-_df['add']).days
        _df['day']=_day
        if _day%7==0:
            _df['week']=_day/7
        else:
            _df['week']=_day//7+1
    except:
        pass
    return _df
data=data.apply(get_data,axis=1)
day=data[data['name']=='《最美的时候遇见你》']['day'].head(1).values[0]
box=data[data['name']=='《最美的时候遇见你》']['boxoffice'].sum()
mean=round(box/day,6)
s=str(day)+','+str(mean)
f=open('ans0301.dat','w')
f.write(s)
f.close()

转载于:https://my.oschina.net/u/3766838/blog/1605439

你可能感兴趣的:(获取DataFrame第一列的值(无魔法数字))