Копировать IEnumerable, заменяя элемент

Question

Я хочу сделать копию IEnumerable<T>, в которой один элемент по данному индексу был заменен заданным значением.

я определил следующий метод, который делает то, что я хочу:

public static IEnumerable<T> ReplaceAt<T>(this IEnumerable<T> source, T item, int index) 
{ 
    foreach (T before in source.Take(index)) 
    { 
     yield return before; 
    } 

    yield return item; 

    foreach (T after in source.Skip(index + 1)) 
    { 
     yield return after; 
    } 
} 

Однако, в то время как, возможно, легко понять это кажется «неэффективным», чтобы создать два итератора, один из которых пропускает элементы, которые уже приняты первый итератор.

Есть ли лучший способ определить это?

Answer 1

Как насчет:

public static IEnumerable<T> ReplaceAt<T>(this IEnumerable<T> source, T item, int index) 
{ 
    return source.Select((value, i) => index == i ? item : value); 
} 

Answer 2

Не уверен, что эффективно, но вы пробовали это?

public static IEnumerable<T> ReplaceAt<T>(this IEnumerable<T> source, T item, int index) 
{ 
    return source.Select((x, i) => i == index ? item : x); 
} 

Answer 3

Если вы хотите, чтобы сойти с ума, вы можете раскатать foreach вручную:

public static IEnumerable<T> ReplaceAt<T>(this IEnumerable<T> source, T item, int index) 
{ 
    int itemIndex = 0; 
    using(var iter = source.GetEnumerator()) 
    { 
     while(iter.MoveNext()) 
     { 
      yield return itemIndex++ == index ? item : iter.Current; 
     } 
    } 
}