Compute the Indices of the Target Element in Array/List using Py
- 时间:2020-09-13 14:33:25
- 分类:网络文摘
- 阅读:123 次
Given an array (or list), and a target element, find all the indices that the element appears in it. For example,
Array: [1, 2, 3, 4, 5, 5, 6, 7, 8], and find the element 5, which appears in index 4 and 5, thus return [4, 5].
This is a trivial question and most programmers know how to do this using a for loop. For example,
In Python:
1 2 3 4 5 6 | def getIndices(data, element): x = [] for i in range(len(data)): if data[i] == element: x.append(i) return x |
def getIndices(data, element): x = [] for i in range(len(data)): if data[i] == element: x.append(i) return x
Or similarly in Javascript:
1 2 3 4 5 6 7 8 9 | function getIndices(data, element) { let x = []; for (let i = 0; i < data.length; i += 1) { if (data[i] === element) { x.push(i); } } return x; } |
function getIndices(data, element) { let x = []; for (let i = 0; i < data.length; i += 1) { if (data[i] === element) { x.push(i); } } return x; }
However, as a spirit of avoiding loops as much as you can, we can simplify the implementation. In Python, we can use the enumerate and the list comprehension which gives us the following short and concise solution to return the list of the indices for a target element in the array.
1 2 | def getIndices(data, element): return [x for x, y in enumerate(data) if y == element] |
def getIndices(data, element): return [x for x, y in enumerate(data) if y == element]
In Javascript, we could similarly implement a enumerate function, alternatively, we can chain map and filter function.
1 2 3 4 5 6 7 8 | function getIndices(data, element) { let i = -1; return data.map(x => { i ++; // incrementing the index // return index for target element return x == element ? i : false; }).filter(x => typeof x !== 'boolean'); } |
function getIndices(data, element) { let i = -1; return data.map(x => { i ++; // incrementing the index // return index for target element return x == element ? i : false; }).filter(x => typeof x !== 'boolean'); }
The map function maps those target elements with their indices, and others similar set to false – which will be filtered out via filter.
–EOF (The Ultimate Computing & Technology Blog) —
推荐阅读:Tips for Starting a Car-Themed Blog How to Build a Popular, Trustworthy Blog 6 Simple Ways to Promote Your Blog The Secret to Writing More Compelling Blog Titles 5 Habits Of Highly Successful Bloggers 5 Ways to Connect Your Blog Content Writers Google’s Perfect World: A New Technical SEO Framework Boost the Visibility of Your Blog With In-Person Events Are Top 20 Witnesses Voting Each Other? Introducing Witness-voti How to Design a Browser History using Double-ended Queue (deque)
- 评论列表
-
- 添加评论