python 将一个panda Dataframe 拆分为相等的多个 Dataframe 并转换为json

qmelpv7a  于 2023-02-15  发布在  Python
关注(0)|答案(1)|浏览(150)

我的程序的一部分需要某种类型的输入。所以我需要将一个Pandas Dataframe 拆分成n个 Dataframe ,并将其转换为没有该输入头的json。例如,我有一个这样的 Dataframe

`import pandas as pd
myDf = pd.DataFrame(['banana', 'apple', 'watermelon','grapes','cherry', 'blueberry'], columns=['fruit'])`

看起来像这样

`        fruit
0      banana
1       apple
2  watermelon
3      grapes
4      cherry
5   blueberry`

我需要一个类似于l1的程序输出

`import json
l1 = [['banana','apple', 'watermelon'], ['grapes', 'cherry', 'blueberry']]
print(json.dumps(l1, indent = 4))`

它应该如下所示,因为它应该排除标头"fruit"

`[
    [
        "banana",
        "apple",
        "watermelon"
    ],
    [
        "grapes",
        "cherry",
        "blueberry"
    ]
]`

我把 Dataframe 弄散了

fruitList = np.array_split(myDf, 2)

然后在for循环中我尝试了

`json.loads(fruitList_iter.reset_index(drop=True).to_json(orient='values'))`

但没有达到我的预期效果。如果能提供帮助,我将不胜感激。

f2uvfpb9

f2uvfpb91#

将底层numpy数组用于reshape

out = json.dumps(myDf['fruit'].to_numpy().reshape(2, -1).tolist(), indent=4)

输出:

[
    [
        "banana",
        "apple",
        "watermelon"
    ],
    [
        "grapes",
        "cherry",
        "blueberry"
    ]
]

相关问题