Python csv thay thế hàng

Thư viện pandas và csv là những mô-đun phổ biến nhất để xử lý dữ liệu CSV. Bài viết này sẽ thảo luận về cách sử dụng các thư viện này để cập nhật tệp CSV

Chúng ta sẽ xem cách chỉnh sửa tệp CSV bằng cách chèn một hàng, cập nhật một [hoặc nhiều hàng] dựa trên [các] giá trị cột hoặc xóa một [hoặc nhiều hàng] dựa trên các giá trị của một cột đã cho. Chúng ta cũng sẽ xem cách chèn một cột vào tệp CSV

Các ví dụ hiển thị trong bài viết này sẽ sử dụng tệp CSV có tên “employees. csv” với nội dung như sau

ID,Position,Salary,Joining Date,Grade
2001,Junior web developer,5600,8/27/2020,AB
2006,Accountant,7200,7/22/2019,BA
3014,Senior Data Scientist,7800,5/1/2013,AA
1946,HR Manager,6200,9/9/2021,DB
1947,HR Manager 3,6250,1/3/2016,DC

Tệp có sáu cột và năm hàng và được lưu trong cùng thư mục với tập lệnh chứa mã của chúng tôi

Ghi chú. Mỗi đoạn mã được hiển thị sẽ bắt đầu bằng “nhân viên. csv” chứa nội dung trên

Chèn một hàng vào tệp CSV

Trong thư viện csv, chúng ta có thể dễ dàng ghi một hàng vào tệp CSV bằng csv. đối tượng nhà văn []. Trong trường hợp này, chúng ta cần biết các cột trên dữ liệu gốc và thứ tự của chúng

1

2

3

4

5

6

7

8

9

10

11

12

13

# Nhập thư viện csv

nhập csv

# Mở “nhân viên. csv"  ở chế độ chắp thêm ["a"]

# Nó có nghĩa là dòng sẽ được thêm vào sẽ đi đến hàng cuối cùng

với mở["nhân viên. csv", "a"] as infile:

    # Tạo đối tượng người viết cho csv

    người viết = csv. người viết[infile]

    # Dữ liệu chúng tôi muốn ghi vào tệp CSV

    dòng = [2008, "Procurement officer",6100,"2/8/2011","CA"]

    # Ghi hàng vào tệp CSV.

    # Lưu ý rằng dòng này cập nhật tệp tại chỗ

    nhà văn. nhà văn[dòng]

Đầu ra [nội dung của nhân viên. csv sau khi cập nhật]

ID,Position,Salary,Joining Date,Grade
2001,Junior web developer,5600,8/27/2020,AB
2006,Accountant,7200,7/22/2019,BA
3014,Senior Data Scientist,7800,5/1/2013,AA
1946,HR Manager,6200,9/9/2021,DB
1947,HR Manager 3,6250,1/3/2016,DC
2008,Procurement officer,6100,2/8/2011,CA

Đoạn mã trên mở tệp CSV ở chế độ nối thêm [“a”]. Điều đó có nghĩa là hàng chúng tôi đang viết được thêm vào làm hàng cuối cùng

Ngoài ra, bạn có thể sử dụng gấu trúc để thêm hàng vào tệp CSV. Có hai cách để làm điều đó

• Mở tệp CSV ở chế độ chắp thêm, sau đó cập nhật tệp hoặc,
• Sử dụng gấu trúc. Khung dữ liệu. chức năng định vị [ ]

Phương thức đầu tiên hoạt động như open[] và csv. nhà văn [] trong ví dụ trước

Chúng tôi cần mở tệp CSV ở chế độ nối thêm và cập nhật nó bằng một Khung dữ liệu mới chứa [các] hàng mà chúng tôi muốn thêm. Đây là một ví dụ

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

nhập gấu trúc dưới dạng pd

# Dòng chúng tôi muốn thêm vào tệp CSV

dòng = [[1901, "Procurement officer 2",6100,"12/8/2010","CA"],

[1902, "HR",7000 và “Không” nếu ngược lại. Chúng tôi sử dụng NumPy để tận dụng việc vector hóa thuộc tính tính toán

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

nhập gấu trúc dưới dạng pd

nhập numpy as np

df = pd. read_csv["employees. csv"]

#Nêu điều kiện

điều kiện = [df["Salary"]>7000, df["Salary"]