Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Lấy chỉ từ viết hoa từ DataFrame trong Pandas

Để lấy chỉ từ viết hoa, dùng regex. Module re được nhập.

Nhập thư viện:

import re
import pandas as pd

Tạo DataFrame:

data = [['computer', 'mobile phone', 'ELECTRONICS', 'electronics'],
        ['KEYBOARD', 'charger', 'SMARTTV', 'camera']]
dataFrame = pd.DataFrame(data)

Trích xuất từ viết hoa:

for i in range(dataFrame.shape[1]):
    for ele in dataFrame[i]:
        if bool(re.match(r'\w*[A-Z]\w*', str(ele))):
            print(ele)

Ví dụ

import re
import pandas as pd

data = [['computer', 'mobile phone', 'ELECTRONICS', 'electronics'],
        ['KEYBOARD', 'charger', 'SMARTTV', 'camera']]
dataFrame = pd.DataFrame(data)

print("Dataframe...\n", dataFrame)
print("\nDisplaying only capital words...\n")

for i in range(dataFrame.shape[1]):
    for ele in dataFrame[i]:
        if bool(re.match(r'\w*[A-Z]\w*', str(ele))):
            print(ele)

Đầu ra

Dataframe...
       0              1           2          3
0  computer  mobile phone  ELECTRONICS  electronics
1  KEYBOARD      charger    SMARTTV      camera

Displaying only capital words...
KEYBOARD
ELECTRONICS
SMARTTV