Microsoft MVP성태의 닷넷 이야기
.NET Framework: 836. C# - Python range 함수 구현 [링크 복사], [링크+제목 복사]
조회: 944
글쓴 사람
홈페이지
첨부 파일

C# - Python range 함수 구현

Python의 range 함수는 [from, to) 범위의 수를 반환하는데 이런 용도로 C#에서 쉽게 사용할 수 있는 코드가 System.Linq에서 제공하는 Enumerable.Range 메서드입니다.

foreach (int i in Enumerable.Range(1, 10))
{
}

하지만, 아쉽게도 step 기능이 없습니다. 그래서 이를 감안해 다음과 같이 별도로 구현해야 합니다.

C# equivalent of Python's range with step?
; https://stackoverflow.com/questions/43897078/c-sharp-equivalent-of-pythons-range-with-step

using System;
using System.Collections.Generic;

public static class EnumerableUtilities
{
    public static IEnumerable<int> RangePython(int start, int stop, int step = 1)
    {
        if (step == 0)
            throw new ArgumentException("Parameter step cannot equal zero.");

        if (start < stop && step > 0)
        {
            for (var i = start; i < stop; i += step)
            {
                yield return i;
            }
        }
        else if (start > stop && step < 0)
        {
            for (var i = start; i > stop; i += step)
            {
                yield return i;
            }
        }
    }
}

그런데, 구현을 하다 보니 기왕이면 int뿐만 아니라 double 같은 타입에도 사용하고 싶습니다. 문제는 이러한 목적으로 사용할 수 있는 마땅한 제네릭 제약이 없다는 것입니다. 그나마 가장 가까운 것이 C# 7.3의 unmanaged 제약인데, 그렇다고 해서 산술 연산을 지원하는 것은 아니므로 각각의 연산자 사용에서 오류가 발생합니다.

public static IEnumerable<T> Range<T>(T start, T stop, T step = 1) where T : unmanaged
{
    if (step == 0)
        throw new ArgumentException("Parameter step cannot equal zero.");

    if (start < stop && step > 0)
    {
        for (var i = start; i < stop; i += step)
        {
            yield return i;
        }
    }
    else if (start > stop && step < 0)
    {
        for (var i = start; i > stop; i += step)
        {
            yield return i;
        }
    }
}

최대한 오류를 피하기 위해 갖은 인터페이스를 활용해 구현한다면 ^^; 딱 2개의 코드를 제외하고는 다음과 같은 식으로 끌고 가볼 수는 있습니다.

public static IEnumerable<T> RangePython<T>(T start, T stop, T step) where T : unmanaged, IComparable, IEquatable<T>
{
    if (step.Equals(default(T)))
        throw new ArgumentException("Parameter step cannot equal zero.");

    if (start.CompareTo(stop) < 0 && step.CompareTo(0) > 0)
    {
        for (var i = start; i.CompareTo(stop) < 0; i += step)
        {
            yield return i;
        }
    }
    else if (start.CompareTo(stop) > 0 && step.CompareTo(0) < 0)
    {
        for (var i = start; i.CompareTo(stop) > 0; i += step)
        {
            yield return i;
        }
    }
}

그렇습니다. 문제는 바로 저 "+" 연산자입니다. "+" 연산에 대한 인터페이스 제약은 없으므로 어떻게 해볼 도리가 없습니다. 그래도.... 역시 억지로 구현해 보면 가능한 해결책은 있습니다.

public static IEnumerable<T> RangePython<T>(T start, T stop, T step) where T : unmanaged, IComparable, IEquatable<T>
{
    if (step.Equals(default(T)))
        throw new ArgumentException("Parameter step cannot equal zero.");

    if (start.CompareTo(stop) < 0 && step.CompareTo(default(T)) > 0)
    {
        for (var i = start; i.CompareTo(stop) < 0;)
        {
            yield return i;
            Add(ref i, step);
        }
    }
    else if (start.CompareTo(stop) > 0 && step.CompareTo(default(T)) < 0)
    {
        for (var i = start; i.CompareTo(stop) > 0;)
        {
            yield return i;
            Add(ref i, step);
        }
    }
}

private unsafe static void Add<T>(ref T value, T step) where T : unmanaged
{
    switch (value)
    {
        // unsafe를 사용하지 않으면 박싱이 발생하고,
        case short s:
            short sStep = (short)(object)step;
            s += sStep;
            value = (T)(object)s;
            break;

        case int i:
            int iStep = (int)(object)step;
            i += iStep;
            value = (T)(object)i;
            break;

        // unsafe를 사용하면 코드가 좀 복잡해지고!
        case double _:
            {
                double* pStep = (double *)&step;
                fixed (T* pValue = &value)
                {
                    double* pTargetDouble = (double*)pValue;
                    *pTargetDouble += *pStep;
                }
            }
            break;

        default:
            throw new NotSupportedException(value.GetType().ToString());
    }
}

어쨌든 저 정도만 구현해도 다음과 같은 식으로는 사용할 수 있습니다.

static void Main(string[] args)
{
    foreach (short val in Extension.RangePython<short>(1, 2, 1))
    {
        Console.WriteLine(val);
    }

    foreach (double val in Extension.RangePython<double>(0.1, .2, 0.001))
    {
        Console.WriteLine(val);
    }

    foreach (int val in Extension.RangePython<int>(1, 10, 1))
    {
        Console.WriteLine(val);
    }
}

그렇지만, 아무리 봐도 타입별로 어수룩한 중복 코드가 case 문으로 나열되는 Add 문이 영 눈에 밟힙니다. ^^ 그렇다면 이제 남은 가장 최선의 방법은 바로 dynamic을 사용하는 것입니다. ^^

// Is there a constraint that restricts my generic method to numeric types?
// https://stackoverflow.com/questions/32664/is-there-a-constraint-that-restricts-my-generic-method-to-numeric-types
public static IEnumerable<T> RangePython<T>(T start, T stop, T step)
{
    dynamic dStart = start;
    dynamic dStop = stop;
    dynamic dStep = step;

    if (dStep == 0)
        throw new ArgumentException("Parameter step cannot equal zero.");

    if (dStart < dStop && dStep > 0)
    {
        for (var i = dStart; i < dStop; i += dStep)
        {
            yield return i;
        }
    }
    else if (dStart > dStop && dStep < 0)
    {
        for (var i = dStart; i > dStop; i += dStep)
        {
            yield return i;
        }
    }
}

그나마 가장 낫죠? ^^

(첨부 파일은 이 글의 예제 코드를 포함합니다.)




[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[연관 글]





[최초 등록일: ]
[최종 수정일: 5/22/2019 ]

Creative Commons License
이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.
by SeongTae Jeong, mailto:techsharer@outlook.com

비밀번호

댓글 쓴 사람
 




1  2  3  4  5  6  7  8  9  10  11  12  13  14  [15]  ...
NoWriterDateCnt.TitleFile(s)
11858정성태3/27/20191255VC++: 129. EXE를 LoadLibrary로 로딩해 PE 헤더에 있는 EntryPoint를 직접 호출하는 방법파일 다운로드1
11857정성태3/26/20191309VC++: 128. strncpy 사용 시 주의 사항(Linux / Windows)
11856정성태3/25/20191153VS.NET IDE: 134. 마이크로소프트의 CoreCLR 프로파일러 리눅스 예제를 Visual Studio F5 원격 디버깅하는 방법 [1]파일 다운로드1
11855정성태3/25/20191624개발 환경 구성: 436. 페이스북 HTTPS 인증을 localhost에서 테스트하는 방법
11854정성태3/25/2019870VS.NET IDE: 133. IIS Express로 호스팅하는 사이트를 https로 접근하는 방법
11853정성태3/24/20191169개발 환경 구성: 435. 존재하지 않는 IP 주소에 대한 Dns.GetHostByAddress/gethostbyaddr/GetNameInfoW 실행이 느리다면? - 두 번째 이야기
11852정성태3/20/20191262개발 환경 구성: 434. 존재하지 않는 IP 주소에 대한 Dns.GetHostByAddress/gethostbyaddr/GetNameInfoW 실행이 느리다면?파일 다운로드1
11851정성태3/19/20191462Linux: 8. C# - 리눅스 환경에서 DllImport 대신 라이브러리 동적 로드 처리
11850정성태3/18/20191085.NET Framework: 813. C# async 메서드에서 out/ref/in 유형의 인자를 사용하지 못하는 이유
11849정성태3/18/20191617.NET Framework: 812. pscp.exe 기능을 C#으로 제어하는 방법파일 다운로드1
11848정성태3/17/2019987스크립트: 14. 윈도우 CMD - 파일이 변경된 경우 파일명을 변경해 복사하고 싶다면?
11847정성태3/17/20191706Linux: 7. 리눅스 C/C++ - 공유 라이브러리 동적 로딩 후 export 함수 사용 방법파일 다운로드1
11846정성태3/15/20191528Linux: 6. getenv, setenv가 언어/운영체제마다 호환이 안 되는 문제
11845정성태3/15/20192008Linux: 5. Linux 응용 프로그램의 (C++) so 의존성 줄이기(ReleaseMinDependency) [3]
11844정성태5/22/20191968개발 환경 구성: 434. Visual Studio 2019 - 리눅스 프로젝트를 이용한 공유/실행(so/out) 프로그램 개발 환경 설정 [1]파일 다운로드1
11843정성태3/14/20191065기타: 75. MSDN 웹 사이트를 기본으로 영문 페이지로 열고 싶다면?
11842정성태5/3/20191087개발 환경 구성: 433. 마이크로소프트의 CoreCLR 프로파일러 예제를 Visual Studio CMake로 빌드하는 방법 [1]파일 다운로드1
11841정성태3/13/2019900VS.NET IDE: 132. Visual Studio 2019 - CMake의 컴파일러를 기본 g++에서 clang++로 변경
11840정성태3/13/2019967오류 유형: 526. 윈도우 10 Ubuntu App 환경에서는 USB 외장 하드 접근 불가
11839정성태3/12/20191412디버깅 기술: 124. .NET Core 웹 앱을 호스팅하는 Azure App Services의 프로세스 메모리 덤프 및 windbg 분석 개요 [2]
11838정성태5/9/20192094.NET Framework: 811. (번역글) .NET Internals Cookbook Part 1 - Exceptions, filters and corrupted processes [1]파일 다운로드1
11837정성태10/14/20196791기타: 74. 도서: 시작하세요! C# 7.3 프로그래밍 [10]
11836정성태10/12/20191565오류 유형: 525. Visual Studio 2019 Preview 4/RC - C# 8.0 Missing compiler required member 'System.Range..ctor' [1]
11835정성태3/5/20191720.NET Framework: 810. C# 8.0의 Index/Range 연산자를 .NET Framework에서 사용하는 방법 및 비동기 스트림의 컴파일 방법 [1]파일 다운로드1
11834정성태3/4/20191191개발 환경 구성: 432. Visual Studio 없이 최신 C# (8.0) 컴파일러를 사용하는 방법
11833정성태5/14/20191513개발 환경 구성: 431. Visual Studio 2019 - CMake를 이용한 공유/실행(so/out) 리눅스 프로젝트 설정파일 다운로드1
1  2  3  4  5  6  7  8  9  10  11  12  13  14  [15]  ...