pandas 如何仅将 Dataframe 列输出到csv？

wwwo4jvm 于 2022-12-28 发布在其他

关注(0)|答案(1)|浏览(123)

用例：在CLI上的实用函数中打高尔夫球，我不能让它变得复杂。
我只需要查看一个二进制格式的大文件的列名，而不是列名加上第一个数据行。
在我当前的实现中，我必须编写繁琐的命令来查看大文件的第一行：

my-tool peek -n 1 huge-file.parquet | head -n 1 | tr ',' '\n' | less

我想说的是：

my-tool peek --cols huge-file.parquet | tr ',' '\n' | less

或

my-tool peek --cols -d '\n' huge-file.parquet | less

在python中，我目前使用以下机制来生成csv：

out = StringIO()
df.to_csv(out)
print(out.getvalue())

有没有DataFrame-ish的方法可以通过to_csv(...)或类似的简单技术只将列输出到out？

pandas

来源：https://stackoverflow.com/questions/74891119/how-does-only-output-only-dataframe-columns-to-csv

1条答案

按热度按时间

daolsyd01#

像这样的东西？

import pandas as pd
import numpy as np

if __name__ == "__main__":
    # some fake data for setup
    np.random.seed(1)
    df = pd.DataFrame(
        data=np.random.random(size=(5, 5)),
        columns=list("abcde")
    )

    out = df.columns.to_frame(name="columns")
    out.to_csv("file.csv", index=False)
    print(out)

columns
a       a
b       b
c       c
d       d
e       e

赞(0）回复(0）举报 2022-12-28

我来回答

pandas 如何仅将 Dataframe 列输出到csv？

1条答案

相关问题

热门标签

最新问答