添加列并逐行設(shè)置值

import tushare as ts
import time
import pandas as pd

def statisticsfordayofweek(code):
    marketdata = ts.get_k_data(code)
    for index, row in marketdata.iterrows():
        marketdata.loc[index, 'dayofweek'] = time.strftime('%A', time.strptime(row.date, "%Y-%m-%d"))
    downdata = marketdata[marketdata.close < marketdata.open]
    dayweekgroup = downdata['dayofweek'].groupby(
        downdata['dayofweek'].map(lambda  x: x[0:4]))\
        .count()
    dayweekgroup.plot(kind="bar")
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.rcParams['axes.unicode_minus'] = False
    plt.show()

從DataFrame獲取特定列數(shù)據(jù)

data = pd.read_csv('./data/optdigits.tra', header=None)
#注意iloc的用法，這里表示x為所有行，索引為0~63，共64列的數(shù)據(jù)。
#python3中，使用data[range(64)].values，會(huì)報(bào)slice不正確的異常
x, y = data.iloc[:,range(64)].values, data.iloc[:, 64].values
images = x.reshape(-1, 8, 8)
y = y.ravel().astype(np.int)

完全構(gòu)建新的DataFrame，并添加數(shù)據(jù)

dfresult = pd.DataFrame(columns=('domicile',
                                         'universe',
                                         'rawname',
                                         'suggestid',
                                         'legalname',
                                         'Similarity%'))
dfresult.loc[0] = {'domicile': 'UK',
                           'universe':'ETF',
                           'rawname': 'IL Bright Start College Savings (Advisor) Advisor Age Based 15-17 Yrs Port',
                           'suggestid':'F123456ABC',
                            'legalname':'IL Bright Start College Savings (Advisor) Advisor Age Based',
                            'Similarity%':'81.75'}

導(dǎo)出csv時(shí)，解決中文亂碼問題

dfresult.to_csv('./output/result_%s.csv' % searchdate, encoding='utf_8_sig')

解決merge過(guò)程中，數(shù)據(jù)類型不匹配問題

dfresult = pd.DataFrame(columns=('documentid',
                                     'effectivedatehat'))
    indexfordf = 0
    for docid in docdate:
        dfresult.loc[indexfordf] = {'documentid': docid,
                                    'effectivedatehat': docdate[docid]
                                    }
        indexfordf += 1

    dfresult['documentid'] = dfresult['documentid'].apply(int)
    dfsample = pd.read_csv('./output/sample/rawsupplementdocwitheffectivedate.csv', encoding='utf-8')
    dfsample['documentid'] = dfsample['documentid'].apply(int)
    dfmerge = pd.merge(dfsample, dfresult, on = ['documentid'], how='left')