Computer >> Máy Tính >  >> Lập trình >> Python

Thêm cột mới vào DataFrame hiện có trong Pandas

Khung dữ liệu Pandas là cấu trúc dữ liệu hai chiều, tức là dữ liệu được căn chỉnh theo kiểu bảng trong các hàng và cột. Nó có thể được tạo bằng python dict, list và series, v.v. Trong bài viết này, chúng ta sẽ xem cách thêm một cột mới vào khung dữ liệu hiện có. Trong ví dụ dưới đây, chúng tôi đang chuyển đổi chuỗi gấu trúc thành Khung dữ liệu của một cột, đặt tên cột là Month_no.

Ví dụ

import pandas as pd
s = pd.Series([6,8,3,1,12])
df = pd.DataFrame(s,columns=['Month_No'])
print (df)

Đầu ra

Chạy đoạn mã trên cho chúng ta kết quả sau:

   Month_No
0       6
1       8
2       3
3       1
4       12

sử dụng hàm insert ()

Chúng ta có thể sử dụng hàm insert () của pandas để chèn cột vào vị trí được chỉ định bởi chỉ mục của nó. Dưới đây, chúng tôi thêm Số ngày trong tháng dưới dạng cột vào DataFrame gấu trúc hiện có ở vị trí chỉ mục 1.

Ví dụ

import pandas as pd
s = pd.Series([6,8,3,1,12])
df = pd.DataFrame(s,columns=['Month_No'])

# Insert the new column at position 1.
df.insert(1,"No_of_days",[30,31,31,31,31],True)
print (df)

Đầu ra

Chạy đoạn mã trên cho chúng ta kết quả sau -

   Month_No No_of_days
0       6       30
1       8       31
2       3       31
3       1       31
4       12      31

Sử dụng hàm gán ()

Hàm gán ()

Ví dụ

import pandas as pd
s = pd.Series([6,8,3,1,12])
df = pd.DataFrame(s,columns=['Month_No'])

# Insert a column at the end
df = df.assign(No_of_days = [30,31,31,31,31])

print (df)

Đầu ra

Chạy đoạn mã trên cho chúng ta kết quả sau -

   Month_No No_of_days
0       6       30
1       8       31
2       3       31
3       1       31
4       12      31