I have a list1 = [1,2,3,4,5,6,7,8,9,0]
. I want to take out one element "4", then split the remaining list with np.array_split(list1,5)
, I will get [array([1, 2]), array([5, 6]), array([7, 8]), array([ 9, 10]), array([11])]
as result. When I try to convert it into an pandas Data Frame, the out put result would be as:
index | 0 | 1 |
---|---|---|
0 | 1 | 2.0 |
1 | 5 | 6.0 |
2 | 7 | 8.0 |
3 | 9 | 10.0 |
4 | 11 | NaN |
But I want to get the result as just one column data frame without NaN
value at the end.
Any suggestion with this matter would be appreciated.
CodePudding user response:
Put your array into a dict and create your dataframe from that:
list1 = [1,2,3,4,5,77,8,9,0]
x = np.array_split(list1, 5)
df = pd.DataFrame({'column': x})
Output:
>>> df
column
0 [1, 2]
1 [3, 4]
2 [5, 77]
3 [8, 9]
4 [0]