Python：データフレームの文字列の与えられた位置の次の文字を取得する関数

Question

データフレーム内の文字列の別の文字の位置を与える次の文字を取得します。私たちが文字列のすべての文字を閲覧したなら、私は次の行に行くべきです。

これを行うには、以下の関数を書いています。そのデータフレームのための

def get_char(df, y, z): 
    if z < len(df[0][y])-1: 
     return df[0][y][z+1] 
    elif z == len(df[0][y])-1: 
     if y < len(df[0])-1: 
      return df[0][y+1][0] 

：

ar = np.array(["aba", "bcb", "zab"]) 
df = pd.DataFrame(ar) 

I

print get_char(df, 1, 2) 

は私にz

と

print get_char(df, 2, 2) 

を与えた場合

は何も返さないはずです。私の関数ではNoneを返します。

私ははるかに簡単な方法でそれを行うことができます。

私のデータフレームには1つの列しかありません。

Answer 1

import numpy as np 
import pandas as pd 

ar = np.array(["aba", "bcb", "zab"]) 
df = pd.DataFrame(ar) 

def get_char(df, y, z): 
    a = ''.join(df.iloc[y:y+2,0]) 
    try: 
     return a[z+1] 
    except: 
     return None 

print(get_char(df, 1, 2)) 
# z 
print(get_char(df, 2, 2)) 
# None