向CSV编写一种热编码

时间:2019-04-30 15:16:14

标签: python pandas csv

我目前正在尝试使用df.to_csv将一种热编码写入csv,但是当我将其写入原始状态时,数据就处于原始状态,而不是一种热编码。我尝试重命名df,但出现元组错误。代码如下。

Code:

df = pd.read_csv('Filename.txt')
df.columns = ['Date','b1','b2','b3','b4','b5','b6']
df = df.set_index('Date')

reversed_df = df.iloc[::-1]

n=5
print('Game')
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
print(reversed_df.drop(df.index[n:-n]),("\n"))

BallOne = pd.get_dummies(reversed_df.b1)
BallTwo = pd.get_dummies(reversed_df.b2)
BallThree = pd.get_dummies(reversed_df.b3)
BallFour = pd.get_dummies(reversed_df.b4)
BallFive = pd.get_dummies(reversed_df.b5)
BallSix = pd.get_dummies(reversed_df.b6)
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
pd.set_option('display.width', None)
print(pd.concat([BallOne, BallTwo, BallThree, BallFour, BallFive, BallSix], keys = ['B1', 'B2', 'B3', 'B4', 'B5', 'B6'], axis=1),("\n"))
df.to_csv('Filename.csv')

1 个答案:

答案 0 :(得分:0)

在这里您将df序列化,而没有写出pd.concat 操作,你最好做

df = pd.concat([BallOne, BallTwo, BallThree, BallFour, BallFive, BallSix], keys = ['B1','B2', 'B3', 'B4', 'B5', 'B6'],axis=1)

而且应该会变得更好