1

KM 推定にライフラインを使用することに問題があります。従業員の種類 (フルタイム、パートタイムなど) と呼ばれる変数の列があり、KM の見積もりをグループ化し、CSVファイルに出力したいと考えています。ここにスニペットがあります：

worker_types = df['Emp_Status'].unique() 

for i, worker_type in enumerate(worker_types): 
    ix = df['Emp_Status'] == worker_type 
    kmf.fit(T[ix], C[ix]) 
    kmf.survival_function_['worker'] = worker_type 
    #print kmf.survival_function_ 
    kmf.surviva

l_function_.to_csv('C:\Users\Downloads\test.csv')

print 関数を使用すると、 ごとに KM 推定値の各反復が取得されworker_typeます。ただし、csvファイルにエクスポートしようとすると、ワーカー タイプの最新の見積もりしか得られません。

私はライフラインのドキュメントを読み、さまざまなレベルのプロットの例を見ましたが、それを にエクスポートする方法がわかりませんcsv。

4

1 に答える 1

0

ループの先頭でファイルを追加モードで開き、各行を追加できます。次に例を示します。

worker_types = df['Emp_Status'].unique() 
with open('C:/Users/Downloads/test.csv', 'a') as fou:
    for i, worker_type in enumerate(worker_types): 
        ix = df['Emp_Status'] == worker_type 
        kmf.fit(T[ix], C[ix]) 
        kmf.survival_function_['worker'] = worker_type 
        if i == 0:
            kmf.survival_function_.to_csv(fou) # write header on first iteration
        else:
            kmf.survival_function_.to_csv(fou, header=False)

補足: Python 内の Windows パスにバックスラッシュを使用しないでください。代わりにスラッシュを使用してください。

于 2016-07-26T14:08:30.030 に答える