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

猜你喜欢