csvファイルを使用して列に名前を付ける方法

Question

列名をcsvファイルから呼び出すデータセットに入れたいとします。 私のコードは完全に正常に動作します。

DF2 = pd.DataFrame(data=DF1,index=SKU, 
    columns=['USER1','USER2','USER3','USER4','USER5','USER6']) 

6列。

私は約50 +の列を持っていて、リストを入力するのではなくローカルに保存されているUSERIDという名前のcsvファイルから列名を読みたいと思います。どうしたらいいですか？

次のコードは、

USERID = pd.read_csv("C:\EVALUATE\USERID.csv") 
DF2 = pd.DataFrame(data=DF1,index=SKU, columns=USERID) 

任意の提案を動作しませんでしたか？

Answer 1

ファイルはCSV形式であることを持っています - あなたは可能性があり、単純にpipe the column names from standard inputsplitting the input lines、その後chaining them togetherによって空白で区切られた単語の流れのように：？

import fileinput 
import itertools 

USERID = itertools.chain(*(line.split() for line in fileinput.input())) 
DF2 = pd.DataFrame(data=DF1,index=SKU, columns=USERID) 

次に、あなたがファイルを持っていることを考えるとUSERID.txtこれは次のようになります。

USER1 USER2 
USER3 
USER4 USER5 
USER6 

... python DF2.py < USERID.txtPOSIX shellまたはin a Windows shellおよびlist(USERID)のいずれかが['USER1','USER2','USER3','USER4','USER5','USER6']のようになります。

この場合の唯一の欠点は、空白を含む列名を使用できないことですが、その要件を満たすためにこのコードとデータ形式を変更するのは簡単です。 （プリント `の出力は何

import itertools 

with open("C:\EVALUATE\USERID.txt", "r") as USERID_instream: 
    USERID = itertools.chain(*(line.split() for line in USERID_instream)) 
DF2 = pd.DataFrame(data=DF1,index=SKU, columns=USERID)