将一个热编码写入cs

2024-04-19 02:11:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我目前正试图写一个热编码到csv,使用df.to_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')

Tags: columnscsvtononedfgetdisplayfilename