编写一个 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
广告