繁体   English   中英

在 Python 中的 CSV 文件顶部插入一行

[英]Insert a row at top of CSV file in Python

我想使用 python 在我的 csv 文件的顶部追加一行。 我有 4 列需要添加。 到目前为止,这是我的代码:

rows= ('A','B','C','D')

fd = open('file.csv','a')
fd.write(rows)
fd.close()

但是,这有两个问题:我收到一条错误消息,说“预期是字符缓冲区对象”,我确信这与我的变量“行”有关。

第二个问题是我相信这只会将它附加到底部,而我需要它在顶部。

任何帮助,将不胜感激。

你这里似乎有两个问题:

  1. 您收到一条错误消息,提示“需要字符缓冲区对象”。

    这是因为您只能将字符串或字符数组写入文件,而元组不是这些东西(即使它是字符串或字符的元组)。 您必须首先将元组转换为字符串。 一种简单的方法是使用str(('A', 'B', 'C', 'D'))repr(('A', 'B', 'C', 'D')) 如果这对您不起作用,那么最好提取每个组件并从中形成一个字符串,例如

    a = '' for c in ('A', 'B', 'C', 'D'): a += c + ' '
  2. 您想附加到文本文件的顶部而不是底部。 不幸的是,您不能简单地做到这一点。 有关完整说明,请参见此处 解决这个问题的方法是将整个文件作为一个字符串读入,将所需的文本插入到它的开头,然后将其全部重写到一个文件中。

对于这么简单的事情来说有点矫枉过正,但我​​发现拥有一个处理诸如操作之类的电子表格的类非常有帮助。 这是一个围绕独立行的简单方法。

class Table():
    def __init__(self):# instanciates an empty table
        self.rows = []
    def push(self,row): # adds a row to the top of the table
        self.rows = [row]+self.rows
    def que(self,row): #adds a row to the bottom of the table
        self.rows = self.rows+[row]
    def remRowAt(self,i): # Removes a row from the table at a given index
        if(i>=0 and i<len(self.rows)):
            self.rows=self.rows[0:i]+self.rows[i+1:len(self.rows)]
        else:print("index error removing at:"+str(i))
    def addRowAt(self,i,row): #Adds a row at a given index
        if(i>=0 and i<= len(self.rows)):
            self.rows = self.rows[0:i]+[row]+self.rows[i:len(self.rows)]
        else:print("index error adding at:"+str(i))
    def prt(self,delim): # returns the table in the form of a string.
        s =""
        for row in self.rows:
            for entry in row:
                s+= str(entry)+delim
            s=s[0:len(s)-1]+"\n"
        s=s[0:len(s)-1]
        return(s)
    def read(self,s,delim):
        for k in s.split("\n"):
            self.que(k.split(delim))

t = Table()
t.push(['a','b','c','d'])
t.push([1,2,3,4])
t.que(['check','my','work'])
t.remRowAt(1)
t.addRowAt(2,[2,3,4,5])
print(t.prt(","))
copyT = Table()
copyT.read(t.prt(","),',')
print(">")
print(copyT.prt("\t"))

屈服

1,2,3,4
check,my,work
2,3,4,5
>
1   2   3   4
check   my  work
2   3   4   5

为了解决这个问题,您注意到 prt 方法返回一个字符串而不是一个列表,允许将其传递给 file.write() 方法。

为什么会出错?

当需要"character buffer object"时,您正在传递一个要write的元组。 实际上,这意味着它需要一个字符串。

我建议使用 python csv.writer类来帮助你。 https://docs.python.org/2/library/csv.html#csv.writer

写入文件顶部。

也许这个答案有帮助:

Python f.write() 在文件开头?

我不是一个有经验的程序员,但我在顶部添加行的逻辑是这样的:

  1. 对 CSV 数据进行排序并将其倒置(我认为 Pandas 具有排序功能)

    您可能需要添加一个数字为 0-n 的列(我的意思是序列号),然后您可以根据降序对数据进行排序。

  2. 然后追加行 >> 你知道它将被追加到底部。

  3. 按照递增的数字再次排序。

  4. 现在删除您添加的列。

这样,您底部的数据将到达顶部!

我希望这有帮助。

你可以试试这个:

import csv

row = ['A','B','C','D']

with open(filename, 'r') as readFile:
    rd = csv.reader(readFile)
    lines = list(rd)
    lines.insert(0, row)

with open(filename, 'w',newline='') as writeFile:
    wt = csv.writer(writeFile)
    wt.writerows(lines)

readFile.close()
writeFile.close()

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM