編寫一個 Python 程式,生成隨機的十行兩列母音字母。如果兩行都匹配相同的母音字母,則列印匹配列的索引和計數。
假設你有一個數據框,
col1 col2 0 o e 1 e e 2 i u 3 e o 4 i i 5 u o 6 e a 7 u o 8 a u 9 e a
匹配索引和計數的結果是:
index is col1 col2 1 e e 4 i i count is 2
解決方案
為了解決這個問題,我們將遵循以下步驟:
定義一個數據框
使用以下方法比較第一個和第二個匹配的索引值:
df.iloc[np.where(df.col1==df.col2)])
使用以下步驟查詢匹配列的總計數:
len(df.iloc[np.where(df.col1==df.col2)])
示例
讓我們看看以下實現以更好地理解:
import pandas as pd import numpy as np df = pd.DataFrame({'col1': np.random.choice(['a','e','i','o','u'], 10), 'col2': np.random.choice(['a','e','i','o','u'],10) }) print("DataFrame is \n", df) print("index is \n",df.iloc[np.where(df.col1==df.col2)]) print("count is \n", len(df.iloc[np.where(df.col1==df.col2)]))
輸出
col1 col2 0 o e 1 e e 2 i u 3 e o 4 i i 5 u o 6 e a 7 u o 8 a u 9 e a index is col1 col2 1 e e 4 i i count is 2
廣告