I have been testing some issues I've had with Pandas. My end goal here is to add data to a .csv. While figuring out ways to change a .csv, I settled on this method:
import pandas
data = pandas.read_csv('path/to/my/script/test.csv')
data.iat[1,1] = 'DataHere'
data.to_csv('path/to/my/script/test.csv', index=False, header=False)
This code worked somewhat correctly. DataHere goes to the second row and second column, which is correct (because [0,0] is the first row and column. Note: it's not the normal x,y coordinates, it's more like y,x).
test.csv before code (6x6):
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
test.csv after code (6x5):
yes,yes,yes,yes,yes,yes
yes,DataHere,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
It gets rid of the lowermost row for some reason! So I did some messing with the parameters of pandas.read_csv('path/to/my/script/test.csv') to fix this problem, and got this:
data = pandas.read_csv('path/to/my/script/test.csv', nrows=6, skip_blank_lines=False)
I added nrows=6 to make it read 6 rows, although I do intend to make it higher in the future. I added skip_blank_lines=False because I want to be able to add data to blank cells.
When I ran this new code (after changing the csv to its previous 6x6 state), it didn't help. It still erases the 6th row.
import pandas
data = pandas.read_csv('path/to/my/script/test.csv', nrows=6, skip_blank_lines=False)
data.iat[1,1] = 'DataHere'
data.to_csv('path/to/my/script/test.csv', index=False, header=False)
I also tried data.iat[6,3] = 'DataHere' instead of data.iat[1,1] = 'DataHere', which returned this error:
IndexError: index 6 is out of bounds for axis 0 with size 5
This shows that not only it is erasing the last row, but that it cannot add data to a blank cell. To make sure that it was the fault of this line: data = pandas.read_csv('path/to/my/script/test.csv', nrows=6, skip_blank_lines=False), I put print(data) in the line immediately after it and got this output (plus the previously stated errors). There should be a 5th row of 'yes' there. So my two problems are:
- Deletion of a row.
- Not being able to add data to a blank cell.
CodePudding user response:
pandas.read_csv('path/to/my/script/test.csv') uses the first row as a header row. Your test.csv does not have a header row. So it is likely that the first row (data row) in test.csv is being read as a header row. Giving you 5 data rows and not 6 as you expect.
This could be happening
sim_csv = io.StringIO(
'''yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes
yes,yes,yes,yes,yes,yes'''
)
data = pd.read_csv(sim_csv)
print(data)
yes yes.1 yes.2 yes.3 yes.4 yes.5
0 yes yes yes yes yes yes
1 yes yes yes yes yes yes
2 yes yes yes yes yes yes
3 yes yes yes yes yes yes
4 yes yes yes yes yes yes
Then when you write out the CSV with to_csv(header=None) you lose that first row of data.
To get around this you could do:
pandas.read_csv('path/to/my/script/test.csv', header=None)
Adding a row (cell to a new row)
You can add a row (cell to a row) by using the index. For example if you had:
yes yes.1 yes.2 yes.3 yes.4 yes.5
0 yes yes yes yes yes yes
1 yes yes yes yes yes yes
2 yes yes yes yes yes yes
3 yes yes yes yes yes yes
4 yes yes yes yes yes yes
Then you could do: (note that this is .at and not .iat)
df.at[5,'yes'] = 'yes'
Which will give you:
yes yes.1 yes.2 yes.3 yes.4 yes.5
0 yes yes yes yes yes yes
1 yes yes yes yes yes yes
2 yes yes yes yes yes yes
3 yes yes yes yes yes yes
4 yes yes yes yes yes yes
5 yes NaN NaN NaN NaN NaN
