Python 正規表現について

Question

Pythonで入力した数が整数でも有理数でも計算できるプログラムを書きたいと思っています。

import re

class Rat:
    n = 0
    d = 0

    def __init__(self, n, d):
        self.n = n
        self.d = d

    def gcd(a, b):
        if(a % b == 0):
            return b
        else:
            return Rat.gcd(b, a % b)

    def __repr__(self):
        if(self.d / Rat.gcd(self.n, self.d) != 1):
            return '{}/{}'.format(int(self.n / Rat.gcd(self.n, self.d)), int(self.d / Rat.gcd(self.n, self.d)))
        elif(self.d / Rat.gcd(self.n, self.d) == 1):
            return '{}'.format(int(self.n / Rat.gcd(self.n, self.d)))

    def __add__(self, other):
        return Rat((self.n * other.d) + (self.d * other.n), (self.d * other.d))

    def __sub__(self, other):
        return Rat((self.n * other.d) - (self.d * other.n), (self.d * other.d))

    def __mul__(self, other):
        return Rat((self.n * other.n), (self.d * other.d))

    def __truediv__(self, other):
        return Rat((self.n * other.d), (self.d * other.n))


def is_int(arg):
    """
    int型かどうか調べる関数
    """
    try:
        int(arg)
        return True

    except ValueError:
        return False

def input_number():
    """
    標準入力で分数あるいは整数を受け取り、全て分数に変換して返す関数
    """
    while(True):
        msg = input("Number is fraction or integer? Enter (f or i):")

        if(msg == "i"):
            num = input("Enter integer:")
            if(is_int(num)):
                num = Rat(int(num), 1)
                break
            else:
                print("ValueError : not integer")
                continue

        elif(msg == "f"):
            tmp = input("Enter fraction like 3/5:")
            print(re.match("//", tmp))
            if(re.match("//", tmp)):
                tmp_list = re.split("//", tmp)
                print(tmp_list)
                if((is_int(tmp[0]) and is_int(tmp[1]))):
                    num = Rat(int(tmp[0]), int(tmp[1]))
                    print(num)
                    break
                else:
                    print("ValueError : not fraction")
                    continue

        elif(msg == "quit"):
            exit()

    return num

def main():
    num1 = input_number()
    num2 = input_number()

    print()
    print("Arithmetic of num1 & num2")
    print("({}) + ({}) = {}".format(num1, num2, num1 + num2))
    print("({}) - ({}) = {}".format(num1, num2, num1 - num2))
    print("({}) * ({}) = {}".format(num1, num2, num1 * num2))
    print("({}) / ({}) = {}".format(num1, num2, num1 / num2))
    print()

if __name__ == "__main__":
    main()

しかし、実行すると分数の"/"で分割して分数の分母分子を区別したいのに、コマンドラインで"/"を入力しても認識されません。おそらくは正規表現の部分でNoneと返って来ているのでその部分が怪しいと思うのですが・・・。

Yuji-no-MacBook-Air:tmp_file YujiNarita$ python hw0423.py
Number is fraction or integer? Enter (f or i):f
Enter fraction like 3/5:23/34
None
Number is fraction or integer? Enter (f or i):quit
Yuji-no-MacBook-Air:tmp_file YujiNarita$ python hw0423.py
Number is fraction or integer? Enter (f or i):i
Enter integer:6
Number is fraction or integer? Enter (f or i):i
Enter integer:8

Arithmetic of num1 & num2
(6) + (8) = 14
(6) - (8) = -2
(6) * (8) = 48
(6) / (8) = 3/4

Yuji-no-MacBook-Air:tmp_file YujiNarita$ 

なお、上記の通り、整数では問題なく計算できました。また、正規表現を使わず("/"をコマンドラインで入力し忘れるなどのErrorを無視して)書いた場合は有理数でも問題なく計算できたのでclass Ratの部分は間違っていないかと思われます。

もし記載不足などございましたらご指摘いただけると幸いです。また、コードの書き方などもご指摘いただければと思います。

Answer 1

tmp = input("Enter fraction like 3/5:")
print(re.match("//", tmp))

間違いが二つあります。
0. re.matchは、先頭一致を判定するものです。
0. スラッシュはメタ文字としての意味を持ちませんから、単に'/'でいいです。

単に『スラッシュが含まれるかどうか』を判定したいのであれば、正規表現は不要です。

tmp = input("Enter fraction like 3/5:")
print('/' in tmp)

なお

fractionsモジュールが標準で用意されているので、こちらを使うと楽です。

Comments

[north_redwings]

先頭だったんですね・・・。勉強不足でした。
()でキャプチャして成功しました、ありがとうございました。

Answer 2

import re

class Rat:
    n = 0
    d = 0

    def __init__(self, n, d):
        self.n = n
        self.d = d

    def gcd(a, b):
        if(a % b == 0):
            return b
        else:
            return Rat.gcd(b, a % b)

    def __repr__(self):
        if(self.d / Rat.gcd(self.n, self.d) != 1):
            return '{}/{}'.format(int(self.n / Rat.gcd(self.n, self.d)), int(self.d / Rat.gcd(self.n, self.d)))
        elif(self.d / Rat.gcd(self.n, self.d) == 1):
            return '{}'.format(int(self.n / Rat.gcd(self.n, self.d)))

    def __add__(self, other):
        return Rat((self.n * other.d) + (self.d * other.n), (self.d * other.d))

    def __sub__(self, other):
        return Rat((self.n * other.d) - (self.d * other.n), (self.d * other.d))

    def __mul__(self, other):
        return Rat((self.n * other.n), (self.d * other.d))

    def __truediv__(self, other):
        return Rat((self.n * other.d), (self.d * other.n))


def is_int(arg):
    """
    int型かどうか調べる関数
    """
    try:
        int(arg)
        return True

    except ValueError:
        return False

def input_number():
    """
    標準入力で分数あるいは整数を受け取り、全て分数に変換して返す関数
    """
    while(True):
        msg = input("Number is fraction or integer? Enter (f or i):")

        if(msg == "i"):
            num = input("Enter integer:")
            if(is_int(num)):
                num = Rat(int(num), 1)
                break
            else:
                print("ValueError : not integer")
                continue

        elif(msg == "f"):
            tmp = input("Enter fraction like 3/5:")
            m = re.match("(\d+)(/)(\d+)", tmp)
            if m:
                num = Rat(int(m.group(1)), int(m.group(3)))
                break
            else:
                print("ValueError : not fraction")
                continue

        elif(msg == "quit"):
            exit()

    return num

def main():
    num1 = input_number()
    num2 = input_number()

    print()
    print("Arithmetic of num1 & num2")
    print("({}) + ({}) = {}".format(num1, num2, num1 + num2))
    print("({}) - ({}) = {}".format(num1, num2, num1 - num2))
    print("({}) * ({}) = {}".format(num1, num2, num1 * num2))
    print("({}) / ({}) = {}".format(num1, num2, num1 / num2))
    print()

if __name__ == "__main__":
    main()